Tensorflow GPU设置:PyCharm上的CUDA错误

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.在PyCharm之外的 iPython 中(即使用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_...如果您希望将来创建的文件具有此功能,也可以放在那里).请注意,您需要保存此配置,否则当您关闭PyCharm时它不会粘住.