Ubuntu 22.04.5深度学习环境配置
时间: 2025-06-17 19:02:05 AIGC 浏览: 34
### 设置Ubuntu 22.04.5 LTS上的深度学习开发环境
#### 安装Miniconda
为了管理不同的Python环境,建议先安装Miniconda。通过Miniconda可以更方便地管理和隔离不同项目的依赖项。
```bash
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
```
完成安装后重启终端或运行`source ~/.bashrc`使更改生效[^2]。
#### 配置CUDA和cuDNN
对于GPU加速的支持,需要安装合适的CUDA工具包以及配套的cuDNN库。考虑到兼容性和性能优化,推荐按照目标硬件选择相应版本的CUDA。
下载并执行CUDA安装文件:
```bash
sudo apt-get update && sudo apt-get install -y gcc make
wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run
chmod +x cuda_11.7.1_515.65.01_linux.run
sudo ./cuda_11.7.1_515.65.01_linux.run --silent --toolkit
echo 'export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc
```
接着安装匹配版本的cuDNN,通常可以通过NVIDIA官网获取预编译好的二进制文件,并解压到CUDA目录下。
#### 创建Conda虚拟环境并安装PyTorch
利用Conda创建一个新的Python环境来保持项目独立性:
```bash
conda create -n pytorch_env python=3.9
conda activate pytorch_env
```
之后可以直接通过pip命令安装特定于所选CUDA版本的PyTorch轮子(wheel),而不需要完整版的CUDA Toolkit:
```bash
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
```
这里选择了与之前安装的CUDA 11.7相适应的PyTorch构建[^3]。
#### 测试安装
最后验证一切正常工作:
```python
import torch
print(torch.cuda.is_available())
print(torch.__version__)
```
如果输出显示CUDA可用并且打印出了正确的PyTorch版本号,则表明配置成功。
阅读全文
相关推荐




















