Ubuntu 22.04 安装 CUDA 和 cuDNN 完整步骤
Ubuntu 22.04 安装 CUDA 和 cuDNN 完整步骤
一、准备工作
-
安装 NVIDIA 驱动
- 打开“软件和更新”（Software & Updates），选择“附加驱动”标签页，安装推荐的专有驱动版本（如
nvidia-driver-535）。 - 或通过命令行安装：
sudo ubuntu-drivers autoinstall sudo reboot - 验证驱动安装：
nvidia-smi # 显示显卡信息即成功
- 打开“软件和更新”（Software & Updates），选择“附加驱动”标签页，安装推荐的专有驱动版本（如
-
安装依赖项
sudo apt update && sudo apt install gcc g++ make -y
二、安装 CUDA 11.7
-
下载 CUDA 安装包
访问 CUDA Toolkit Archive，选择 CUDA 11.7 的 runfile（local） 版本，复制下载命令：wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run -
执行安装
sudo sh cuda_11.7.1_515.65.01_linux.run- 输入
accept接受协议。 - 取消勾选驱动安装（已安装驱动时需取消，避免冲突）。
- 勾选
CUDA Toolkit并确认安装路径（默认/usr/local/cuda-11.7）。
- 输入
-
配置环境变量
echo 'export PATH=/usr/local/cuda-11.7/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc -
验证 CUDA 安装
nvcc -V # 显示 CUDA 版本即成功
三、安装 cuDNN 8.6.0
-
下载 cuDNN 安装包
访问 cuDNN 下载页面，选择 cuDNN 8.6.0 for CUDA 11.x 的 Local Installer for Ubuntu22.04 (Tar) 版本。 -
解压并安装
tar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include sudo cp cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* -
验证 cuDNN 安装
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # 输出版本号即成功
四、常见问题解决
-
CUDA 版本冲突
- 若安装多个 CUDA 版本，通过修改软链接切换版本：
sudo rm /usr/local/cuda sudo ln -s /usr/local/cuda-11.7 /usr/local/cuda
- 若安装多个 CUDA 版本，通过修改软链接切换版本：
-
cuDNN 权限问题
- 若出现权限错误，手动修改文件夹权限：
sudo chmod 666 /usr/local/cuda/include sudo chmod 666 /usr/local/cuda/lib64
- 若出现权限错误，手动修改文件夹权限：
-
验证 PyTorch 支持
import torch print(torch.cuda.is_available()) # 输出 True 即成功
五、注意事项
- 驱动与 CUDA 版本匹配：通过
nvidia-smi显示的 CUDA Version 是显卡支持的最高版本，实际安装的 CUDA 版本需不高于此值。 - cuDNN 兼容性：确保 cuDNN 版本与 CUDA 版本对应（如 cuDNN 8.6.0 需搭配 CUDA 11.x）。
- 离线安装：若需离线安装，提前下载
.deb或.run文件，并注册 NVIDIA 开发者账号以获取下载权限。
