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! 则证明安装成功
本文转载自: https://blog.csdn.net/qq_32381353/article/details/136712326
版权归原作者 thatyimjune 所有, 如有侵权,请联系我们删除。
版权归原作者 thatyimjune 所有, 如有侵权,请联系我们删除。