1、去官网下载需要的CUDNN(https://developer.nvidia.com/cuDNN),需要邮箱注册。
2、解压
tar zxvf cudnn-8.0-linux-x64-v6.0.tgz
3、删除原来的CUDNN
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
4、cd 到刚才解压的文件夹进行安装
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
5、cd到/usr/local/cuda/lib64/文件夹下,建立软链接(注意版本号换成你自己的)
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.6.0.21
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so
sudo ldconfig