tensorflow - 从源代码构建 TensorFlow 后，看到 libcudart.so 和 libcudnn 错误

Question

我正在从源代码构建 TensorFlow。构建似乎成功了；但是，当我的 TensorFlow 程序调用 import tensorflow 时，会出现以下一个或两个错误:

• 导入错误:libcudart.so.8.0:无法打开共享对象文件:没有这样的文件或目录
• 导入错误:libcudnn.5:无法打开共享对象文件:没有这样的文件或目录

Answer 1

首先，出现以下错误:

ImportError: libcudart.so.8.0: cannot open shared object file: No such file or directory

确保您的LD_LIBRARY_PATH包括您的lib64目录中安装 cuda 包的路径。您可以通过添加 export 来做到这一点在你的 .bashrc 中。对于奥马尔来说，它看起来如下:

I fixed this just adding the cuda path to my .bashrc

export LD_LIBRARY_PATH=/usr/local/cuda/lib64/

<小时/>

对我来说，我必须遵循奥马尔的路线，而且: export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64/因为我有两个涉及cuda的目录(可能不是最好的)。

<小时/>

第二，您确定安装了cuDNN吗？请注意，这与常规 cuda 包不同。您需要注册，然后从以下页面下载并安装软件包: https://developer.nvidia.com/cudnn

<小时/>

第三，我也遇到了同样的问题:

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

事实证明没有libcudnn.5在我的/usr/local/cuda/lib64或/usr/local/cuda-8.0/lib64目录。不过，我确实有一个libcudnn.so.6.*文件。为了解决这个问题，我创建了一个软链接(soft link):

ln -s libcudnn.so.6.* libcudnn.so.5

在我的/usr/local/cuda/lib64目录。现在一切都适合我。如果您已有cuDNN，您的目录可能会有所不同，还有你的libcudnn.so.6.*可能是不同的版本，所以请检查一下。