web-dev-qa-db-fra.com

Erreur TensorFlow (GPU) libcudart.so.7.5 - Impossible d'ouvrir le fichier d'objet partagé

Installation de TensorFlow avec GPU sur Ubuntu 14.04. Assez nouvelle dans l'environnement Ubuntu/UNIX, donc beaucoup de choses que je ne comprends pas. Lors de la recherche de ce problème, j'ai rencontré des cas d'impossibilité d'importer CUDA 7.0 lorsqu'ils ont installé CUDA 7.5 (car TensorFlow ne prend pas en charge 7.5). CUDA 7.0 est installé sur mon ordinateur, mais il semble être à la recherche de la version 7.5 - pourquoi? J'ai dans mon fichier .bashrc les lignes suivantes:

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

Mais je vois toujours le problème ci-dessous ... Quelqu'un a un conseil?

>>> 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
ubuntutensorflow
7
Chimi

Vous devez installer tensorflow à partir de SOURCE pour configurer les paramètres. L'installation par pip ou par easy_install ne fonctionnera pas. 

J'ai le même problème après la mise à niveau de tensorflow de 0.6.0 à 0.7.1 en utilisant pip.

1
Winter

J'ai eu le même problème. Avez-vous essayé d'installer CUDA 7.5 aux côtés de 7? Cela a fonctionné pour moi et j'ai changé mes variables PATH pour inclure 7.5 au lieu de 7.0

0
penelopeia

Si vous utilisez les derniers binaires r0.7, d'après mon expérience, ils sont maintenant conçus pour prendre en charge CUDA 7.5 par défaut ... alors que les binaires r0.6 fonctionnaient avec CUDA 7.0

Vous pouvez créer tensorflow à partir de la source pour utiliser CUDA 7.0, voir la documentation ici Configurer la vue canonique des bibliothèques Cuda de TensorFlow

$ ./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
j314erre