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.debsudo dpkg -i libcudnn8-dev_8.9.6.50-1+cuda12.2_amd64.debsudo 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! 则证明安装成功



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。