当前位置:首页 » 《随便一记》 » 正文

linux下cuda安装-ubuntu22.04安装cuda11.8(cuda+cudnn)

22 人参与  2024年10月17日 18:00  分类 : 《随便一记》  评论

点击全文阅读


原文链接:linux下cuda安装-ubuntu22.04安装cuda11.8(cuda+cudnn)

导言

cuda是nvidia提供在N卡上实现并行计算的工具包,cudnn是cuda的运行时库,通过cuda提供的内核函数能够高效的在N卡上跑密集计算程序。
本文在N卡机器上安装好了相应的驱动后,提供安装cuda和cudnn的详细步骤,并运行官方示例程序验证。

Nvidia驱动安装参考链接:linux下nvidia驱动安装-ubuntu22.04安装2060-notebook驱动

cuda下载和安装

在官网找到对应环境的cuda(本文使用linux-x64-ubuntu22.04-runfile),最好使用run脚本本地安装,以保证稳定性。

#wget下载,不同环境的run文件不同wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run#执行安装,同意协议sudo sh cuda_11.8.0_520.61.05_linux.run #注意选项中要取消nvidia-driver选择,因为驱动已经安装了

最好不要使用该脚本中的驱动安装方法,驱动、cuda和cudann的安装可能会存在冲突。本文中ubuntu默认gcc11.4,安装550以上的驱动要求gcc12.3.0,但是安装cuda11.8时又要求gcc11.4。
cuda默认安装在了/usr/local/cuda-11.8/bin

#添加cuda到用户环境export PATH=$PATH:/usr/local/cuda-11.8/binexport LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64

cudnn下载安装

官方引导安装方法
在cudnn中先找到找到对应版本下载

可以在兼容列表中查看cudnn、cuda、驱动的兼容性

sudo apt-get install zlib1g #安装依赖sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb #安装cudnnsudo cp /var/cudnn-local-repo-ubuntu2204-8.9.7.29/cudnn-local-8AE81B24-keyring.gpg /usr/share/keyrings/ #添加共享钥匙,版本8.9.7.29不一定相同sudo apt-get update

查看安装位置信息

dpkg-query -L cudnn-local-repo-ubuntu2204-8.9.7.29

安装deb后会在/var/cudnn文件夹可以看到3个deb:libcudnn,dev,samples,将其都安装完成,其中samples就是官方提供的示例程序

cd /var/cudnn-local-repo-ubuntu2204-8.9.7.29/sudo dpkg -i libcudnn8_8.9.7.29-1+cuda11.8_amd64.deb  libcudnn8-dev_8.9.7.29-1+cuda11.8_amd64.deb libcudnn8-samples_8.9.7.29-1+cuda11.8_amd64.deb

还需要安装 freeimage用于模块测试

sudo apt-get install -y libfreeimage3 libfreeimage-dev

cudnn测试

将/usr/src/cudnn-samples-v8复制到任意工作文件夹中,进入cudnn_samples_v8/mnistCUDNN,执行

cd cudnn_samples_v8/mnistCUDNN make clean& make #确保编译无错./mnistCUDNN 

本文测试时出现了gcc12.3.0版本过高的错误,降回11.4.0后测试通过

卸载

#cuda:执行 cuda-uninstaller 在 /usr/local/cuda-11.8/bin 中#cudnn: 检查sudo dpkg -l | grep cudnn卸载并删除gpg钥匙sudo dpkg -r cudnn-local-repo-ubuntu2204-8.9.7.29 & sudo rm -r /usr/share/keyrings/cudnn-local-repo-ubuntu2204-8.9.7.29/cudnn-local-8AE81B24-keyring.gpgsudo apt-get purge cudnn-local-repo-ubuntu2204-8.9.7.29

点击全文阅读


本文链接:http://zhangshiyu.com/post/173281.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1