python - Tensorflow GPU 设置 : error with CUDA on PyCharm

Question

我在 Python3、MacOSX El Capitan 上安装了 TF 0.8。

当为 TF 运行一个简单的测试代码时，我收到这条消息:

ImportError: dlopen(/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so, 10): 
Library not loaded: @rpath/libcudart.7.5.dylib
  Referenced from: /Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so
  Reason: image not found

我的 .bash_profile 如下:

export PATH=/usr/local/bin:$PATH
export DYLD_LIBRARY_PATH=/Developer/NVIDIA/CUDA-7.5/lib:/usr/local/cuda/lib

• 在/Developer/NVIDIA/CUDA-7.5/lib 中我有一个名为 libcudart.7.5.dylib
的 文件
• 在/usr/local/cuda/lib 中我有一个别名 叫做libcudart.7.5.dylib

我已经尝试了几种 .bash_profile 的排列，但都没有成功。知道哪里出了问题吗？

请注意，我可以成功地将我的 GPU 与 Theano 结合使用，因此没有理由相信 GPU/cuDNN/CUDA 安装可能有问题。

Answer 1

如果您收到此错误，请确保按照 Tensorflow 安装说明中的说明正确安装了 CUDA、cuDNN。请注意您正在安装的 TF、CUDA、cuDNN 版本以及您正在使用的 Python 版本。

文件名、路径等经常变化，因此如果突然出现错误，您可能需要对文件名和路径进行小幅调整。有时其他人很难帮助您，因为您的系统可能有一个非常具体的路径设置/版本，论坛中的人无法理解。

如果您得到的正是我在 OP 中描述的错误，请后退一步并检查:

• PyCharm 中会发生这种情况吗？
• 在 PyCharm 中的 iPython 中会发生这种情况吗？
• 两者都发生了吗？
• 这是在终端的 iPython 中发生的吗？

就我而言，它仅发生在 PyCharm 中。在 iPython PyCharm 之外(使用 Mac“终端”软件)一切正常。但是在 PyCharm 中执行 iPython 时，或者通过 PyCharm 运行测试文件时，我会得到错误。这意味着它与 PyCharm 有关，而不是 Tensorflow 安装。

确保您的 DYLD_LIBRARY_PATH 正确指向 libcudart.7.5.dylib 文件。使用 Finder 导航到那里，进行 Spotlight 搜索并找到文件或其别名。然后将该路径放入您的 .bash_profile。就我而言，这是有效的:

export DYLD_LIBRARY_PATH=/usr/local/cuda/lib

如果您的问题是 PyCharm，则需要进行特定配置。转到 GUI 的右上角，然后单击灰色向下箭头。

选择“编辑配置”。您将看到一个 Environment 选项，您需要在其中单击 ... 框并输入适用于您的情况的 DYLD_LIBRARY_PATH。

请注意，您正在处理的特定文件有一个环境选项(它将在左侧面板中突出显示)和默认值(如果需要，也可以将 DYLD_... 放在那里您创建的 future 文件具有此功能)。请注意，您需要保存此配置，否则当您关闭 PyCharm 时它不会保留。