2016-11-21 4 views
2

Я недавно установил tenorflow с поддержкой GPU на своем Ubuntu 16.04, и мне удалось успешно выполнить некоторые тесты в терминале. Позже я решил, что было бы проще иметь IDE для написания кода на Python, и я бы просто импортировал numpy и tensorflow в IDE и запускал все там. Установив Tensorflow успешно, я не хотел создавать другую версию Anaconda, поэтому я только что установил Spyder2 прямо из программного обеспечения Ubuntu. Однако, когда я пытаюсь импортировать tensorflow в Spyder2Как импортировать GPU-тензор в Spyder IDE (без anaconda)?

import tensorflow as tf 

это не удалось с следующим сообщением:

ImportError: Traceback (most recent call last): 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python  
/__init__.py", line 49, in <module> 
    from tensorflow.python import pywrap_tensorflow 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python  
/pywrap_tensorflow.py", line 28, in <module> 
    _pywrap_tensorflow = swig_import_helper() 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python 
/pywrap_tensorflow.py", line 24, in swig_import_helper_mod = imp.load_module('_pywrap_tensorflow', fp, pathname, description) 

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


Error importing tensorflow. Unless you are using bazel, 
you should not try to import tensorflow from its source directory; 
please exit the tensorflow source tree, and relaunch your python  
interpreter from there. 

Я не мог импортировать NumPy без проблем, но не tensorflow. Я новичок в Linux; что мне здесь не хватает? Или мне нужна anaconda, чтобы использовать Spyder и Tensorflow? Благодаря!

+1

Ваш CUDA не установлен или 'LD_LIBRARY_PATH' не установлен правильно. – yuefengz

+0

Я уверен, что CUDA был установлен, так как я могу запустить shadoworflow в командной строке. Я понимаю, что LD_LIBRARY_PATH - это переменная среды - вы говорите, что переменная окружения не установлена ​​должным образом для Spyder для поиска Tensorflow? Если нет, как мне сообщить Spyder, где искать Tensorflow? – Rexroth

+0

Я думаю, вам нужно запустить Spyder с терминала, чтобы он правильно захватил переменные среды (в этом 'LD_LIBRARY_PATH'). –

ответ

1

Эти переменные должны быть установлены, чтобы TensorFlow знать, где найти Cuda:

LD_LIBRARY_PATH=/usr/local/cuda/lib64 
CUDA_PATH=/usr/local/cuda 
Смежные вопросы