在 Jupyter Notebook 中导入 TensorFlow GPU 时出错答案

【问题标题】：Error importing TensorFlow GPU in Jupyter Notebook在 Jupyter Notebook 中导入 TensorFlow GPU 时出错
【发布时间】：2017-02-16 11:58:37
【问题描述】：

我正在尝试导入 TensorFlow，但最终出现此错误。

ImportError                               Traceback (most recent call last)
/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/__init__.py in <module>()
     60     sys.setdlopenflags(_default_dlopen_flags | ctypes.RTLD_GLOBAL)
---> 61     from tensorflow.python import pywrap_tensorflow
     62     sys.setdlopenflags(_default_dlopen_flags)

/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.py in <module>()
     27             return _mod
---> 28     _pywrap_tensorflow = swig_import_helper()
     29     del swig_import_helper

/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.py in swig_import_helper()
     23             try:
---> 24                 _mod = imp.load_module('_pywrap_tensorflow', fp, pathname, description)
     25             finally:

/Users/Aakaash/miniconda3/lib/python3.5/imp.py in load_module(name, file, filename, details)
    241         else:
--> 242             return load_dynamic(name, filename, file)
    243     elif type_ == PKG_DIRECTORY:

/Users/Aakaash/miniconda3/lib/python3.5/imp.py in load_dynamic(name, path, file)
    341             name=name, loader=loader, origin=path)
--> 342         return _load(spec)
    343 

ImportError: dlopen(/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so, 10): Library not loaded: @rpath/libcudart.8.0.dylib
  Referenced from: /Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so
  Reason: image not found

During handling of the above exception, another exception occurred:

ImportError                               Traceback (most recent call last)
<ipython-input-7-a649b509054f> in <module>()
----> 1 import tensorflow

/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/__init__.py in <module>()
     22 
     23 # pylint: disable=wildcard-import
---> 24 from tensorflow.python import *
     25 # pylint: enable=wildcard-import
     26 

/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/__init__.py in <module>()
     70 for some common reasons and solutions.  Include the entire stack trace
     71 above this error message when asking for help.""" % traceback.format_exc()
---> 72   raise ImportError(msg)
     73 
     74 # Protocol buffers

ImportError: Traceback (most recent call last):
  File "/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/__init__.py", line 61, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module>
    _pywrap_tensorflow = swig_import_helper()
  File "/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow', fp, pathname, description)
  File "/Users/Aakaash/miniconda3/lib/python3.5/imp.py", line 242, in load_module
    return load_dynamic(name, filename, file)
  File "/Users/Aakaash/miniconda3/lib/python3.5/imp.py", line 342, in load_dynamic
    return _load(spec)
ImportError: dlopen(/Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so, 10): Library not loaded: @rpath/libcudart.8.0.dylib
  Referenced from: /Users/Aakaash/miniconda3/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so
  Reason: image not found


Failed to load the native TensorFlow runtime.

See https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md#import_error

for some common reasons and solutions.  Include the entire stack trace
above this error message when asking for help.

我可以毫无问题地导入 TensorFlow 的 CPU 版本。我还可以在从终端启动的 Python 控制台中导入 TensorFlow。在控制台和 Jupyter Notebook 中，python 可执行文件的路径是相同的（sys.executable 在控制台和 Jupyter Notebook 中给出了相同的输出）。

TensorFlow 版本 - 1.0.0

Python 版本 - 3.5.2

Cuda 版本 - 8.0

【问题讨论】：

这似乎是一个安装问题，我们在 github 问题页面上进行了跟踪。请在那里提出问题。根本问题是您的环境无法找到 cuda 库。请适当设置LD_LIBRARY_PATH。您可以在 _pywrap_tensorflow.so 对象上运行 ldd 以检查您是否正确。
好的。我将在 Github 上提交问题。你能告诉我如何正确设置 LD_LIBRARY_PATH 吗？我是新手，我不知道该怎么做。
请关注install guide。它有一个指向 NVIDIA 文档的指针。
谢谢。我会遵守的。
补充一点，我让 tensorflow gpu 在从终端启动的 python 控制台中正常工作。所以我认为我的安装很好。但是在木星内部我也得到了一个错误。我在 ubuntu 上。

标签： python tensorflow jupyter-notebook

【解决方案1】：

看起来错误是在我设置 CUDA 和 CuDNN 时出现的。按照 Nvidia 提供的指南，它告诉我为 Mac OS X 设置 DYLD_LIBRARY_PATH 的路径。不幸的是，这在使用 Jupyter Notebook 时不起作用。我们需要按照 Nvidia 指南针对 Linux 的步骤设置LD_LIBRARY_PATH。这解决了问题。

【讨论】：

【解决方案2】：

在我的情况下，当启动 Jupyter notebook 时，LD_LIBRARY_PATH 被清除，即使它在 Ubuntu 上以 .bashrc 和 .profile 导出。仅使用env VARIABLE=$VARIABLE jupyter notebook 启动笔记本服务器即可解决。

【讨论】：