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.h
sudo 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有区别,因此都操作了一遍