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

来源:互联网 发布:windows xp翻墙 编辑:程序博客网 时间:2024/06/06 04:29

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


阅读全文
0 0