Ubuntu 22.04 安装 CUDA 和 cuDNN 完整步骤_ubuntu22.04安装cuda和cudnn

Ubuntu 22.04 安装 CUDA 和 cuDNN 完整步骤

一、准备工作

1. 安装 NVIDIA 驱动

   • 打开“软件和更新”（Software & Updates），选择“附加驱动”标签页，安装推荐的专有驱动版本（如 nvidia-driver-535）。
   • 或通过命令行安装：

     sudo ubuntu-drivers autoinstallsudo reboot

   • 验证驱动安装：

     nvidia-smi # 显示显卡信息即成功

2. 安装依赖项

   sudo apt update && sudo apt install gcc g++ make -y

二、安装 CUDA 11.7

1. 下载 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

2. 执行安装

   sudo sh cuda_11.7.1_515.65.01_linux.run

   • 输入 accept 接受协议。
   • 取消勾选驱动安装（已安装驱动时需取消，避免冲突）。
   • 勾选 CUDA Toolkit 并确认安装路径（默认 /usr/local/cuda-11.7）。

3. 配置环境变量

   echo \'export PATH=/usr/local/cuda-11.7/bin:$PATH\' >> ~/.bashrcecho \'export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH\' >> ~/.bashrcsource ~/.bashrc

4. 验证 CUDA 安装

   nvcc -V # 显示 CUDA 版本即成功

三、安装 cuDNN 8.6.0

1. 下载 cuDNN 安装包
   访问 cuDNN 下载页面，选择 cuDNN 8.6.0 for CUDA 11.x 的 Local Installer for Ubuntu22.04 (Tar) 版本。

2. 解压并安装

   tar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xzsudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/includesudo cp cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

3. 验证 cuDNN 安装

   cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2# 输出版本号即成功

四、常见问题解决

1. CUDA 版本冲突

   • 若安装多个 CUDA 版本，通过修改软链接切换版本：

     sudo rm /usr/local/cudasudo ln -s /usr/local/cuda-11.7 /usr/local/cuda

2. cuDNN 权限问题

   • 若出现权限错误，手动修改文件夹权限：

     sudo chmod 666 /usr/local/cuda/includesudo chmod 666 /usr/local/cuda/lib64

3. 验证 PyTorch 支持

   import torchprint(torch.cuda.is_available()) # 输出 True 即成功

五、注意事项

• 驱动与 CUDA 版本匹配：通过 nvidia-smi 显示的 CUDA Version 是显卡支持的最高版本，实际安装的 CUDA 版本需不高于此值。
• cuDNN 兼容性：确保 cuDNN 版本与 CUDA 版本对应（如 cuDNN 8.6.0 需搭配 CUDA 11.x）。
• 离线安装：若需离线安装，提前下载 .deb 或 .run 文件，并注册 NVIDIA 开发者账号以获取下载权限。