运行TensorFlow程序时的ImportError:libcudnn

Question

尝试运行TensorFlow程序时遇到以下错误:

ImportError: libcudnn.Version: cannot open shared object file: No such file or director

Answer 1

只需下载cuDNN 5.1并按照步骤(在Ubuntu 16.04上测试,CUDA工具包8.0)

$ tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
$ sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

现在设置Path变量

$ vim ~/.bashrc

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

并做了

有关详细信息,您可以查看此站点

Answer 2

看来TensorFlow现在支持cudNN 6.0.我在使用cudNN 5.1时遇到错误.

使用TensorFlow版本1.3.0.

如果你得到了

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

尝试使用6.0

Answer 3

此时Tensorflow不支持cuDNN 6.如果您的错误是找不到libcudnn.so.5而且您只安装了cuDNN 6 libcudnn.so.6,那么在引入cuDNN 6支持之前,您必须安装cuDNN 5.有一个开放的错误报告,您可以在Tensorflow Github页面上查看,以了解他们何时开始支持cuDNN 6.

Answer 4

您可能没有安装正确版本的 cuDNN。

要确定 cuDNN 的正确版本，请参阅在 GPU 支持下运行 TensorFlow 的 NVIDIA 要求。