安装cudnn
https://developer.nvidia.com/cudnn下载安装包(需先注册)
解压到cuda安装目录
1 sudo tar xvf cudnn-8.0-linux-x64-v7.tgz -C /usr/local
2 sudo ldconfig
(cuda默认安装在/usr/local/cuda,这个cudnn解压后一级目录是cuda,所以解压到/usr/local与cuda目录合并,-C是存在该目录才复制,cp命令是不存在则创建。使用“解压到”方式复制后的文件不会丢失软连接信息。)
更新cudnn
cuDNN Archive | NVIDIA Developer
选择你要下载的v3或者v5版本
1、解压出一个名为cuda的文件夹,文件夹中有include和lib64两个文件夹
2、删除原来的cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
1
2
3、安装安装需要版本的cudnn,在终端cd到刚解压的cuda文件夹
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
1
2
4、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
1
2
3
4
5、检测
cd /usr/local/cuda/lib64/
ll
1
2
会显示出你已经建立的软链接记录
至此,cudnn版本更新完毕