Ubuntu20.04下deb格式安装CUDA+cudnn

安装CUDA前需要查看设备是否安装好NVIDIA驱动,具体步骤可以查看我的上一篇安装教程。

云服务器Ubuntu20.04配置NVIDIA驱动+CUDA+cudnn+Anaconda

由于使用runfile格式安装的CUDA和cudnn总是报错,尝试各种解决方案均没有作用,无奈之下尝试使用deb格式重新安装,发现问题解决。

CUDA安装

 进入CUDA官网:CUDA Toolkit Archive

选择好自己设备的信息,并选择deb格式安装

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.3.0/local_installers/cuda-repo-ubuntu2004-12-3-local_12.3.0-545.23.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-3-local_12.3.0-545.23.06-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-3-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-3

 按照Base Installer提供的代码进行安装即可。(这是我需要安装的CUDA版本,其他版本按照流程进行安装即可)

 添加系统路径

进入bashrc文件

sudo nano ~/.bashrc	

在最后添加

#将下面两行插入到文件末尾
export PATH=/usr/local/cuda-11.8/bin:$PATH						
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

 Ctrl+X,按Y,按回车

source ~/.bashrc	#更新文件

验证CUDA安装

nvcc -V

 

图为我用上一篇文章的,正常应该显示CUDA12.3 。

 安装cudnn

进入cudnn的官网:cuDNN Archive

选择CUDA的对应的版本和系统对应的版本下载

 我选择Ubuntu20.04 x86_64(Deb)下载

下载后会得到一个压缩包,将其解压,并进入解压后的文件夹内

输入下面命令

sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_amd64.deb

导入CUDA GPG key

sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/

刷新数据

sudo apt-get update

分别安装三个库

sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y

sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y

sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y

x.x.x是对应下载的cudnn版本信息,X.Y是 cuda的大版本和小版本信息

如果找不到需要填写的内容,可以进入/var/cuda-repo-ubuntu2004-12-3-local/路径下查看
 

ls -flt

会出现对应的deb文件,复制版本信息到上面三条命令进行安装。

安装完成后,查找文件路径

whereis cudnn_version.h

使用以下任意一条命令验证安装是否成功:

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 5 MAJOR

将路径替换为你查找到的路径,运行成功会出现如下内容:

Anaconda安装可以参考我的另一篇文章:

 云服务器Ubuntu20.04配置NVIDIA驱动+CUDA+cudnn+Anaconda

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值