ToB企服应用市场:ToB评测及商务社交产业平台

标题: linux下cuda安装-ubuntu22.04安装cuda11.8(cuda+cudnn) [打印本页]

作者: 北冰洋以北 时间: 2024-7-27 17:21
标题: linux下cuda安装-ubuntu22.04安装cuda11.8(cuda+cudnn)
原文链接：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/bin
export 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 #安装cudnn
sudo 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.gpg
sudo apt-get purge cudnn-local-repo-ubuntu2204-8.9.7.29

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)