Ubuntu14.04更换cudnn版本

来源:互联网 发布:网站性能优化方案 编辑:程序博客网 时间:2024/06/01 09:22

tensorflow升级为1.3以后可能出现

ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory

的问题,这是因为tensorflow支持的cudnn版本应不低于6.0.因此需要更换一波版本~

步骤如下:

删除原来的cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.hsudo 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 rm -rf libcudnn.so libcudnn.so.5 #生成软衔接sudo ln -s libcudnn.so.6.0.8 libcudnn.so.5 #生成软链接sudo ln -s libcudnn.so.5 libcudnn.so 

我担心cuda与cuda-8.0有区别,因此都操作了一遍

原创粉丝点击