在Ubuntu 20.04系统下，使用Python 3.8时，为了正确安装CUDA、cuDNN和PyTorch，需要确保它们的版本是相互兼容的。以下是推荐的版本组合及安装步骤： ### 1. CUDA版本选择 - **推荐版本**：CUDA 11.3 或 CUDA 11.7（与PyTorch 1.12及更高版本兼容）。 - **安装方法**： - 从NVIDIA官网下载对应版本的CUDA Toolkit（`.deb`格式）。 - 运行以下命令安装： ```bash sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_*.deb # 以11.3为例 sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub sudo apt-get update sudo apt-get -y install cuda ``` - 验证安装： ```bash nvcc --version ``` ### 2. cuDNN版本选择 - **推荐版本**：cuDNN 8.2（对应CUDA 11.3）或 cuDNN 8.4（对应CUDA 11.7）。 - **安装方法**： - 从NVIDIA官网下载cuDNN的`.deb`包（需注册NVIDIA开发者账号）。 - 运行以下命令安装： ```bash sudo dpkg -i libcudnn8_*.deb sudo dpkg -i libcudnn8-dev_*.deb ``` - 验证安装： ```bash cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 ``` ### 3. PyTorch版本选择 - **推荐版本**：PyTorch 1.12.1（支持CUDA 11.3）或 PyTorch 1.13.1（支持CUDA 11.7）。 - **安装方法**： - 使用pip安装预编译版本（推荐）： ```bash pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 ``` 或（CUDA 11.7）： ```bash pip3 install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 ``` - 验证安装： ```python python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())" ``` ### 注意事项 1. **环境变量**：确保`~/.bashrc`中包含CUDA路径： ```bash export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 运行`source ~/.bashrc`生效。 2. **Python版本**：Ubuntu 20.04默认Python 3.8可通过`python3`命令调用，无需额外配置。 3. **驱动兼容性**：安装CUDA前需确保NVIDIA驱动版本≥450（通过`nvidia-smi`检查）。