tensorflow - 导入错误 : libcudnn when running a TensorFlow program

Question

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

导入错误:libcudnn.Version:无法打开共享对象文件:没有这样的文件或导演

Answer 1

只是 download 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*

现在设置路径变量

$ 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

并做了

有关更多详细信息，您可以查看此 site