今天在安装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 完美解决。

本文介绍了解决TensorFlow1.4.1与cuDNN版本不兼容的问题,通过下载匹配版本的cuDNN lib包并进行正确配置,最终成功运行TensorFlow。
9万+

被折叠的 条评论
为什么被折叠?



