一、下载cudnn
这是下载官方网址https://developer.nvidia.com/cuDNN,没有注册的用户需要注册才可以下载。
选择自己需要的版本进行下载,解压出一个名为cuda的文件夹,文件夹中有include和lib64两个文件夹
二、删除原来的cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
三、安装需要的cudnn
在终端cd到刚解压的cuda文件夹
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
四、建立软链接
cd到/usr/local/cuda/lib64/文件夹下,建立软链接(注意版本号换成你自己的)
sudo chmod +r libcudnn.so.5.0.5
sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig
五、查看版本信息
cat /usr/local/cuda/version.txt #cuda版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 #cudnn版本