2016-03-01 6 views
7

Instalowanie TensorFlow z GPU w systemie Ubuntu 14.04. Dość nowe środowisko Ubuntu/UNIX, więc wiele rzeczy nie dostaję. Podczas przeszukiwania tego problemu natknąłem się na przypadki niemożności zaimportowania CUDA 7.0, gdy zainstalowano CUDA 7.5 (ponieważ TensorFlow nie obsługuje wersji 7.5). Mam zainstalowany CUDA 7.0, ale wygląda na to, że szukam 7.5 - dlaczego tak jest? Mam w pliku .bashrc następujące linie:Błąd TensorFlow (GPU) libcudart.so.7.5 - nie można otworzyć pliku obiektu współdzielonego

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-7.0/lib64" 
export CUDA_HOME=/usr/local/cuda-7.0 

Ale nadal widzę problem poniżej ... Czy ktoś ma jakieś rady?

>>> import tensorflow as tf 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/__init__.py", line 23, in <module> 
    from tensorflow.python import * 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 49, in <module> 
    from tensorflow import contrib 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/contrib/__init__.py", line 23, in <module> 
    from tensorflow.contrib import layers 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/contrib/layers/__init__.py", line 68, in <module> 
    from tensorflow.contrib.layers.python.layers import * 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/contrib/layers/python/layers/__init__.py", line 22, in <module> 
    from tensorflow.contrib.layers.python.layers.initializers import * 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/contrib/layers/python/layers/initializers.py", line 24, in <module> 
    from tensorflow.python.ops import random_ops 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/ops/random_ops.py", line 23, in <module> 
    from tensorflow.python.framework import ops 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 39, in <module> 
    from tensorflow.python.framework import versions 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/versions.py", line 22, in <module> 
    from tensorflow.python import pywrap_tensorflow 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module> 
    _pywrap_tensorflow = swig_import_helper() 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 24, in swig_import_helper 
    _mod = imp.load_module('_pywrap_tensorflow', fp, pathname, description) 
ImportError: libcudart.so.7.5: cannot open shared object file: No such file or directory 

Odpowiedz

1

konieczne zainstalowanie tensorflow od SOURCE skonfigurować ustawienia. Instalacja za pomocą pip lub easy_install nie będzie działać.

Mam ten sam problem po aktualizacji tensorflow z 0.6.0 na 0.7.1 za pomocą pip.

0

Miałem ten sam problem. Czy próbowałeś zainstalować CUDA 7.5 wraz z 7? To zadziałało dla mnie i zmieniłem zmienne PATH na 7,5 zamiast 7.0

+0

Mam problemy z instalacją instalacji CUDA 7.5 iz jakiegoś powodu nie mogę zlokalizować pliku konfiguracyjnego, który powinienem uruchomić, aby określić, której wersji CUDA użyć. Mogę spróbować zainstalować TensorFlow 0.7.0, ponieważ domyślnie jest to 7.0. – Chimi

0

Jeśli używasz najnowszych plików binarnych r0,7, z mojego doświadczenia wynika, że ​​są one teraz domyślnie obsługiwane w CUDA 7.5 ... binarne r0.6 pracował z CUDA 7.0

można budować tensorflow od źródła do pracy z CUDA 7.0 zamiast zobacz dokumentację tutaj Configure TensorFlow's canonical view of Cuda libraries

$ ./configure 
Please specify the location of python. [Default is /usr/bin/python]: 
Do you wish to build TensorFlow with GPU support? [y/N] y 
GPU support will be enabled for TensorFlow 

Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave 
empty to use system default]: 7.0 

Please specify the location where CUDA 7.0 toolkit is installed. Refer to 
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda 

Please specify the Cudnn version you want to use. [Leave empty to use system 
default]: 4.0.4 

Please specify the location where the cuDNN 4.0.4 library is installed. Refer to 
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cudnn-r4-rc/ 

Please specify a list of comma-separated Cuda compute capabilities you want to 
build with. You can find the compute capability of your device at: 
https://developer.nvidia.com/cuda-gpus. 
Please note that each additional compute capability significantly increases your 
build time and binary size. [Default is: \"3.5,5.2\"]: 3.5 

Setting up Cuda include 
Setting up Cuda lib64 
Setting up Cuda bin 
Setting up Cuda nvvm 
Configuration finished 
+0

Tak, widziałem to, ale nie byłem w stanie zlokalizować pliku konfiguracyjnego, aby zmienić używaną bibliotekę. "Rozwiązałem tymczasowo mój problem, instalując TensorFlow 0.7.0, który domyślnie jest ustawiony na CUDA 7.0 ... – Chimi

+0

skrypt konfiguracyjny znajduje się w katalogu źródłowym tensorflow na najwyższym poziomie po zrobieniu klinu git https://github.com/ tensorflow/tensorflow/blob/master/configure – j314erre