安装完成tensorflow后,发现import tensorflow as tf出现这个错误,分析原因可能是libcudnn.so.5的软连接有问题。
解决方案:
切换到该目录下,删除原本的软连接
cd /usr/local/cuda-8.0/lib64/
sudo rm -rf libcudnn.so.5
重新下载cudnn5.1, 网站链接https://developer.nvidia.com/rdp/cudnn-archive
然后直接右键解压,解压后cuda目录下有include和lib64两个文件夹。
复制头文件cudnn.h到/usr/local/cuda-8.0/include/目录下:
sudo cp cuda/include/cudnn.h/usr/local/cuda-8.0/include/
复制库文件lib*到/usr/local/cuda-8.0/lib64/目录下:
sudo cp cuda/lib/lib*/usr/local/cuda-8.0/lib64/
之后关闭终端,重新导入tensorflow成功