Ubuntu 22.04 安装 CUDA 和 cuDNN 及验证
thatyimjune 2024-08-06 10:07:02 阅读 95
Ubuntu 22.04 安装 CUDA 和 cuDNN 过程及注意事项
安装 CUAD 及验证安装 cuDNN验证 cuDNN 安装是否成功
安装 CUAD 及验证
进入终端 <code>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! 则证明安装成功
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。