Ubuntu 22.04 安装 CUDA 和 cuDNN 及验证

安装 CUAD 及验证

• 进入终端 nvidia-smi查看支持的最高 CUDA 版本，我这里最高支持 12.4
• 进入官网 CUDA Toolkit 下载对应版本的 CUDA，我这里下载 12.1.0
• 选择对应版本 runfile，在终端执行对应的命令
• 其余的按照这个链接里执行就可以了
• 最后，测试是否安装成功：nvcc -V

安装 cuDNN

• 官网链接，现在都需要登陆下载，推荐下载 deb 文件
• 下载对应版本的 deb 文件，命令安装sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.6.50_1.0-1_amd64.deb
• 安装完成会显示，后面的操作命令，执行对应的命令就行：sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.6.50/cudnn-local-1998375D-keyring.gpg /usr/share/keyrings/，若没有显示提示命令则执行通用命令：sudo cp /var/cudnn-local-*/cudnn-*-keyring.gpg /usr/share/keyrings/
• 进入文件夹cd /var/cudnn-local-repo-ubuntu2204-8.9.6.50/，安装对应的依赖包，这些依赖包直接下载安装会失败，都包含在deb文件中，进入上面提到的文件夹安装即可：
  • sudo dpkg -i libcudnn8_8.9.6.50-1+cuda12.2_amd64.deb
  • sudo dpkg -i libcudnn8-dev_8.9.6.50-1+cuda12.2_amd64.deb
  • sudo dpkg -i libcudnn8-samples_8.9.6.50-1+cuda12.2_amd64.deb

验证 cuDNN 安装是否成功

• 最后，测试是否安装成功：
  • 进入文件夹cd /usr/src/cudnn_samples_v8，将 cuDNN 示例复制到用户文件 cp -r /usr/src/cudnn_samples_v8/ $HOME

  若没有该文件夹，按照官网教程安装执行验证

  • 编译 cuDNN mnisiCUDNN 示例 make clean && make

  报错没有找到 FreeImage.h 文件，sudo apt-get install libfreeimage-dev安装依赖

  • 运行 mnistCUDNN 示例./mnistCUDNN
  • 返回 Test passed! 则证明安装成功