如何升级已安装的cudnn版本 Ubuntu16.04 Linux (Caffe、TensorFlow)【以cudnn5.1.10到6.0.21为例】

都是偷懒的锅。。。

已安装:cudnn 5.1.10

现需要:cudnn 6.0.21

1.下载新版本cudnn

        你可以在官网下载,当然需要自己login和survey,然后选择download -> I Agree…… -> Archived cuDNN Releases -> 找到匹配你的CUDA的cudnn的需要版本(cudnn 6.0.21)

        如果你比我还懒,这里有下载好的版本:链接: https://pan.baidu.com/s/1Q-NcMe7ZNTXvddIFNjESUg 密码: 7fik

2.删除原来的cudnn

        包括原来解压的cudnn目录、原来/usr/local/cuda-8.0/lib64中的旧版本libcudnn.so、libcudnn.so.x、libcudnn.so.x.x.x

#命令行操作
cd /usr/local/cuda-8.0/lib64/ sudo rm -rf libcudnn.so libcudnn.so.5 libcudnn.so.5.
1.10#删除原有动态文件

3.安装新版本的cudnn

cd /home/isio/cuda/include #你的新版cudnn解压后的目录
sudo cp cudnn.h /usr/local/cuda-8.0/include/ #复制头文件到cuda的include目录
cd /home/isio/cuda/lib64 #你的新版本cudnn解压后的lib目录
sudo cp lib* /usr/local/cuda-8.0/lib64/ #复制动态链接库到你的cuda的目录

cd /usr/local/cuda-8.0/lib64/ #进入cuda的lib64目录
#根据你新版cudnn的文件名生成软衔接,文件名在解压后的目录看
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -s libcudnn.so.6 libcudnn.so

4.检查安装情况

locate libcudnn.so #查看列表情况

5.测试架构可用情况

TensorFlow直接就能用,Caffe需要重新编译。

TensorFlow版本对应问题看这篇博客

Caffe重新编译,需要cd到Caffe目录下,然后用

sudo make all -j4 #全部重新编译caffe


End

猜你喜欢

转载自blog.csdn.net/weixin_38220948/article/details/80048641
今日推荐