ubuntu22.04 安装深度学习环境
时间: 2024-10-14 16:07:45 浏览: 42
在Ubuntu 22.04上安装深度学习环境通常涉及以下步骤,我们将使用Python的深度学习库如TensorFlow和PyTorch为例:
1. **更新系统**:
```sh
sudo apt update && sudo apt upgrade
```
2. **安装依赖**:
- 安装CUDA和cuDNN (GPU支持):
对于CUDA和cuDNN,你需要下载对应的版本,然后通过`dpkg`安装。如果你有GPU,还需安装驱动程序:
```
sudo apt-get install nvidia-driver-cuda
```
- 安装Python3和pip3:
```sh
sudo apt install python3 python3-pip
```
3. **安装Anaconda(可选,用于管理环境)**:
Anaconda是一个流行的数据科学平台,可以创建和管理独立的Python环境,推荐安装:
```sh
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
```
按照提示操作,选择将Anaconda添加到PATH。
4. **创建并激活新环境**(使用Anaconda):
```sh
conda create -n my_dl_env python=3.9
source activate my_dl_env
```
5. **安装特定的深度学习库**:
- TensorFlow:
```sh
conda install tensorflow
```
- PyTorch:
```sh
pip install torch torchvision torchaudio cudatoolkit=11.0 -f https://download.pytorch.org/whl/cu110/torch_stable.html
```
6. **验证安装**:
启动一个新的Python会话,检查安装是否成功:
```sh
python -c "import tensorflow as tf; print(tf.__version__)"
python -c "import torch; print(torch.__version__)"
```
7. **设置计算设备** (如果有GPU):
在Python里,你可以设置设备来利用GPU:
```python
import tensorflow as tf
tf.config.list_physical_devices('GPU')
```
阅读全文