1、下载CUDNN:
下载地址:cuDNN Archive | NVIDIA Developer(下载的时候需要登陆)
找到自己安装的cuda所对应的cuDNN版本,选择cuDNN Library for Linux(x86_64)就可以
2、下载完成后解压文件,将解压出来的“cudnn-11.1-linux-x64-v8.0.5.39”文件夹里面的"cuda"文件夹复制到主目录下面。(这样做的目的是过会安装的时候安装指令比较简单,不用改目录)
3、复制cuDNN内容到cuda相关文件夹内
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4、创建库文件链接
lib64中有两个文件的大小为0,所以必须要建立库文件链接。(这地方很多教程并没有,很关键。要不会报错内容就是libcudnn.so.7的太小读取不了)来源:ubuntu18.04搭建cuda和cudnn教程(完整教程) - 简书
cd /usr/local/cuda-11.1/lib64
sudo ln -sf libcudnn.so.8.0.5 libcudnn.so.7 #8.0.5是下载的cudnn的版本
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig -v
5、测试是否安装成功:
我看了很多教程里面是输入
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
或
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
但是,对于我安装的这个cudnn版本根本不起作用。
最后的解决方方法:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2没反应解决办法_Loss-CSDN博客
进入到解压的cudnn文件夹中,复制cudnn_version.h到cuda文件夹中即可
sudo cp include/cudnn_version.h /usr/local/cuda/include/
再次执行查看版本命令,成功。
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2