cuDNN升级
出错现象
ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory
查看当前版本为5.0
:/usr/local/cuda/lib64
$ ls -l
lrwxrwxrwx 1 root root 13 12月 23 2016 libcudnn.so -> libcudnn.so.5
lrwxrwxrwx 1 root root 17 12月 23 2016 libcudnn.so.5 -> libcudnn.so.5.1.5
-rwxr-xr-x 1 root root 79337624 12月 23 2016 libcudnn.so.5.1.5
需要升级到6.0版本
1、解压出一个名为cuda的文件夹,文件夹中有include和lib64两个文件夹
2、删除原来的cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
3、安装安装需要版本的cudnn,在终端cd到刚解压的cuda文件夹
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
4、建立软链接(注意版本号换成你自己的)
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
5、检测
cd /usr/local/cuda/lib64/
ll
lrwxrwxrwx 1 root root 13 9月 18 18:17 libcudnn.so -> libcudnn.so.6*
lrwxrwxrwx 1 root root 18 9月 18 18:17 libcudnn.so.6 -> libcudnn.so.6.0.21*
-rwxr-xr-x 1 root root 154322864 9月 18 18:15 libcudnn.so.6.0.21*
会显示出你已经建立的软链接记录
至此,cudnn版本更新完毕