如果觉得本篇文章对您的学习起到帮助作用,请 点赞 + 关注 + 评论 ,留下您的足迹
本文主要wls子系统Ubuntu18.04安装cuDNN,安装cudnn坑巨多,因此记录以备日后查看,同时,如果能够帮助到更多人,也不胜荣幸。
下载Debian本地存储库安装包,根据CUDA版本,下载合适的cudnn版本,点击此处下载cudnn。
本人使用的cuda11.6,因此下载cuDNN8.5.0.96 CUDA 11.x这个版本。
进入cudnn-local-repo-ubuntu1804-8.5.0.96_1.0-1_amd64.deb文件所在文件夹,我们将deb文件存放在用户家目录下
cd ~
sudo dpkg -i cudnn-local-repo-${distro}-8.x.x.x_1.0-1_amd64.deb
# sudo dpkg -i cudnn-local-repo-ubuntu1804-8.5.0.96_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/
# sudo cp /var/cudnn-local-repo-ubuntu1804-8.5.0.96/cudnn-local-7B49EDBC-keyring.gpg /usr/share/keyrings/
sudo apt-get update
更新完成后,注意以下信息
进入/var/cudnn-local-repo-ubuntu18.04-8.5.0.96目录,查看目录信息
cd /var/cudnn-local-repo-ubuntu18.04-8.5.0.96
ls
可以看到三个红色deb文件的cuda版本号是11.7,读者安装时要注意看您的信息内容。
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
# sudo apt-get install libcudnn8=8.5.0.96-1+cuda11.7 # 与上面看到的版本号要一致
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
# sudo apt-get install libcudnn8-dev=8.5.0.96-1+cuda11.7
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
# sudo apt-get install libcudnn8-samples=8.5.0.96-1+cuda11.7
为了验证cuDNN已经安装并正常运行,编译位于文件/usr/src/cudnn_samples_v8目录下的mnistCUDNN示例。
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
还记得上面的三个CUDA版本号为11.7的deb文件吗?安装它们,执行:
cd /var/cudnn-local-repo-ubuntu18.04-8.5.0.96
sudo dpkg -i libcudnn8-dev_8.5.0.96-1+cuda11.7_amd64.deb
sudo dpkg -i libcudnn8-samples_8.5.0.96-1+cuda11.7_amd64.deb
sudo dpkg -i libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb
读者根据自己的cuDNN版本号修改。
sudo apt-get install libfreeimage3 libfreeimage-dev
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN
如果您觉得这篇文章对你有帮助,记得 点赞 + 关注 + 评论 三连,您只需动一动手指,将会鼓励我创作出更好的文章,快留下你的足迹吧