【tensorflow_error】tensorflow:importerror错误汇总

来源:互联网 发布:keep健身软件音乐 编辑:程序博客网 时间:2024/06/16 03:50

最近因为很多东西升级,添加各种库,导致import tensorflow出现各种问题,现汇总如下:

tensorflow:importerror:libcusolver.so.8.0:cannot open shared object file:no such file or directory

由于cuda版本的问题,确保你已经安装cuda8.0,你需要将cuda8.0的路径添加到你的环境中。

打开.bashrc文件,

vim .bashrc

添加如下语句:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64"

libcudnn.so.5:cannot open sharedobject file: No such file or directory

libcudnn.so.6:cannot open sharedobject file: No such file or directory

这两种问题,都是因为cudnn与tensorflow版本不对应的问题。

如果你安装的tensorflow是1.3版本,那么需要支持cudnn6.0,所以请安装cudnn6.0。

如果你安装的tensorflow是1.2版本,那么需要cudnn5.0。

所以如果你已经安装了cudnn,则根据相应的版本去安装tensorflow相应的版本即可。

先卸载之前的tensorflow,然后根据需要安装新的版本。

pip uninstall tensorflow-gpupip install tensorflow-gpu==1.2orpip install tensorflow-gpu==1.3
阅读全文
0 0
原创粉丝点击