1.安装cudnn
首先去官网下载你需要的cudnn,下载的时候需要注册账号。选择对应你cuda版本的cudnn下载。
https://developer.nvidia.com/rdp/cudnn-archive
我下载的两个版本是
Download cuDNN v7.6.5 (November 18th, 2019), for CUDA 10.2
需要下载这4个文件
Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 9.2
2.下载完cudnn之后进行解压,cd进入cudnn解压之后的cuda文件目录
tar zxvf cudnn-9.2-linux-x64-v7.6.5.32.tgz
在命令行进行如下操作安装cuDNN库。:
sudo mv cuda/include/cudnn.h /usr/local/cuda-9.2/include/
sudo mv cuda/lib64/libcudnn* /usr/local/cuda-9.2/lib64
sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*
3.查看cudnn的版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
就可以看到我们7.6.5版本安装成功啦
4.安装样例和开发手册
5.验证cuDNN库。复制例程到~用户目录中
cp -r /usr/src/cudnn_samples_v7/ ~
进入到这个文件中
到这里验证成功啦。
现在cuda版本是cuda9.2,cudnn版本是cudnn7.6.5
6.切换回cuda10.2对应的版本
先要切换到cuda10.2的版本,查看我的上一个博客
添加链接描述
然后再安装对应的cudnn,像上面的安装cuda9.2版本的cudnn一样,在/usr/local/下看到cuda-9.2和cuda-10.2两个文件夹(cudnn的安装是把一个.h头文件和几个lib放到cuda的对应目录下面,记得sudo mv的时候写到真实的cuda-10.2这样的路径下,不要写到cuda软链接路径就好,这样不影响版本对应)
到这里就完成了,如果觉得对你有帮助,就关注一下我,谢谢!!!