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
- Merge pull request #20 from MISingularity/r0.11_new warpctc apply gpu implementation — committed to plutoshe/tensorflow by plutoshe 8 years ago
- Merge pull request #20 from fangweili/master make some bazel targets publicly visible — committed to tarasglek/tensorflow by martinwicke 8 years ago
- Add bazel_build.sh Add use_explicit_batch parameter available in OpConverterParams and other places Formatting and make const bool everywhere Enable use_explicit_batch for TRT 6.0 Revise validatio... — committed to pooyadavoodi/tensorflow by deleted user 5 years ago
- Merge pull request #20 from xmos/master Merge with upstream — committed to xmos/tensorflow by keithm-xmos 4 years ago
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
Out of curiosity, have you set your LD_LIBRARY_PATH to your cuda installation’s lib64 directory?