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版本更新完毕




posted on 2017-09-18 18:19  Maddock  阅读(1198)  评论(0编辑  收藏  举报

导航