今天在安装TensorFlow1.4.1,安装完成后运行爆出如上错误,这是因为1.3以后的版本需要cuDNN v6.0做支持了,本机上安装的是cuDNN v8.0。看网上教程说要安装低版本的TensorFlow,作者感觉不用,就做了如下尝试。
解决方法如下:
1.先到英伟达的官网下载cudnn-8.0-linux-x64-v6.0的lib包。下载完后是cudnn-8.0-linux-x64-v6.0.solitairetheme8这样的文件。
2.解压该文件:
cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz
tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
3.复制文件到/usr/local/cuda/lib64文件夹下,并给相应的权限
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
再运行 import tensorflow as tf 完美解决。