tensorflow: CUDA 7.5 fails with pip install and docker (Ubuntu 14.04)

Installing via:

# For GPU-enabled version (only install this version if you have the CUDA sdk installed)
$ pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl

Tried to run the alexnet_benchmark.py and it’s looking for CUDA 7.0 specifically.

I have CUDA 7.5 on my machine.

Full stack:

Traceback (most recent call last):
  File "alexnet_benchmark.py", line 21, in <module>
    import tensorflow.python.platform
  File "/home/awesomebox/anaconda/lib/python2.7/site-packages/tensorflow/__init__.py", line 4, in <module>
    from tensorflow.python import *
  File "/home/awesomebox/anaconda/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 22, in <module>
    from tensorflow.python.client.client_lib import *
  File "/home/awesomebox/anaconda/lib/python2.7/site-packages/tensorflow/python/client/client_lib.py", line 35, in <module>
    from tensorflow.python.client.session import InteractiveSession
  File "/home/awesomebox/anaconda/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 11, in <module>
    from tensorflow.python import pywrap_tensorflow as tf_session
  File "/home/awesomebox/anaconda/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module>
    _pywrap_tensorflow = swig_import_helper()
  File "/home/awesomebox/anaconda/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.0: cannot open shared object file: No such file or directory

Tried the docker install, but the docker image is configured for a particular NVIDIA driver version, and doesn’t work with others. (this is a known issue: docker driver version and system driver version must exactly match)

About this issue

  • Original URL
  • State: closed
  • Created 9 years ago
  • Comments: 54 (12 by maintainers)

Commits related to this issue

Most upvoted comments

For this error ImportError: libcudart.so.7.5: cannot open shared object file: No such file or directory

I have executed this command to solve it sudo ldconfig /usr/local/cuda/lib64

@ebrevdo yes

printenv LD_LIBRARY_PATH
/usr/local/cuda-7.5/lib64

Out of curiosity, have you set your LD_LIBRARY_PATH to your cuda installation’s lib64 directory?