ImportError: libcudart.so.7.5 当我使用带有 python 扩展的 VScode 调试 TensorFlow 程序时？

Question

我已经在我的 Ubuntu 系统上安装了带有 Python 扩展的 VSCode。我还正确安装了 TensorFlow 0.9、CUDA 7.5 和 cuDNN 库。

但是，当我使用 VSCode 调试 tensorflow 程序时，会报如下错误 importing tensorflow (import tensorflow)：

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

我可以确定所有库都已成功安装。因为我可以在 VSCode 中调试和运行 Python 程序，也可以使用 Ubuntu Terminal 运行 tensorflow 程序。

此外，当我从 Ubuntu 终端导入 tensorflow 时，它的输出如下：

I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcurand.so locally

如您所见，它没有使用库“libcudart.so.7.5”。

我还检查了 ~/.bashrc 中的 CUDA 路径，如下所示：

export CUDA_HOME=/usr/local/cuda-7.5
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64
export PATH=$PATH:/usr/local/cuda-7.5/bin

由于我的 CUDA 安装在路径“/usr/local/cuda-7.5”中，并且我还可以在该路径中找到“libcudart.so.7.5”库，所以我认为没有任何错误。

那么问题是什么？为什么从Terminal导入，没有报错，从vscode导入却出现这样的错误？谢谢。

Answer 1

在搜索和阅读了许多网页之后，我终于解决了这个问题。方法如下：

首先，使用以下命令链接 CUDA 库：

sudo ldconfig /usr/local/cuda/lib64

然后，我可以在 VSCode 中导入 tensorflow。但是，仍然存在一个错误，如下所示：

I tensorflow/stream_executor/dso_loader.cc:99] Couldn't open CUDA library libcudnn.so. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_dnn.cc:1562] Unable to load cuDNN DSO

程序虽然可以运行，但似乎程序找不到cuDNN库。然后我发现我的 Ubuntu 系统上没有正确安装 cuDNN 库。我使用以下命令修改了我的安装：

cd ~/cuda
sudo cp include/cudnn.h /usr/include
sudo cp lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*

这里，'~/cuda' 是我的 cuDNN 安装文件的路径。

最后，经过所有这些步骤，我现在可以导入 tensorflow 并在 VSCode 中调试我的程序了。