2017-02-16 9 views
1

Я пытаюсь импортировать TensorFlow, но в конечном итоге с этой ошибкой.Ошибка импорта графического процессора TensorFlow в Jupyter Notebook

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 без проблем. Я также могу импортировать TensorFlow в Python Console, запущенную с терминала. Путь к исполняемому файлу python такой же как в консоли, так и в Jupyter Notebook (sys.executable дал тот же результат в консоли и Jupyter Notebook).

TensorFlow версия - 1.0.0

Python версии - 3.5.2

Cuda версия - 8,0

+0

Это похоже на проблему с установкой, которую мы отслеживаем на нашей странице проблем github. Пожалуйста, зарегистрируйте там проблему. Корневая проблема заключается в том, что ваша среда не может найти библиотеку cuda. Пожалуйста, настройте 'LD_LIBRARY_PATH' соответствующим образом. Вы можете запустить 'ldd' в объекте' _pywrap_tensorflow.so', чтобы проверить, правильно ли вы это сделали. – drpng

+0

Хорошо. Я напишу о проблеме в Github. Не могли бы вы рассказать мне, как правильно настроить LD_LIBRARY_PATH? Я новичок в этом, и я понятия не имею, как это сделать. –

+0

Пожалуйста, следуйте инструкциям по установке (https://www.tensorflow.org/install/install_linux). Он имеет указатель на документацию NVIDIA. – drpng

ответ

0

Похоже, была ошибка, когда я настройка CUDA и CuDNN. Следуя руководству Nvidia, он сказал мне настроить путь для DYLD_LIBRARY_PATH для Mac OS X. К сожалению, это не работает при использовании Jupyter Notebook. Нам нужно следовать инструкциям руководства Nvidia для Linux и настроить LD_LIBRARY_PATH. Это решило проблему.

0

В моем случае при запуске ноутбука Jupyter LD_LIBRARY_PATH был очищен, даже если экспортировал его в .bashrc и .profile на Ubuntu. Только запуск сервера ноутбука с env VARIABLE=$VARIABLE jupyter notebook решил.