Hi,
Currently working on some neural nets stuff with python 3.6.3, until now I was using tensorflow cpu and recently decided to switch to -GPU version. Followed all the steps from official websites in this order:
- Installation of CUDA Toolkit (9.0)
- Installation of cuDNN v7.0
- Installation of libcupti-dev library (7.5.18-0ubuntu1)
- Installation of Tensorflow-GPU
Of course, after step one and two I have verified the installation using ./deviceQuery and ./bandwidthTest, both of them passed correctly. Also verified cuDNN using ./mnistCUDNN, test also passed.
Installed tensorflow-gpu, tried to verify this using sample code from the website (both python3 Shell and IDE Spyder) but unfortunately no luck.
The stack trace from the shell:
Traceback (most recent call last):
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File "/home/albert/anaconda3/lib/python3.6/imp.py", line 243, in load_module
return load_dynamic(name, filename, file)
File "/home/albert/anaconda3/lib/python3.6/imp.py", line 343, in load_dynamic
return _load(spec)
ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/__init__.py", line 24, in <module>
from tensorflow.python import *
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 72, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/home/albert/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File "/home/albert/anaconda3/lib/python3.6/imp.py", line 243, in load_module
return load_dynamic(name, filename, file)
File "/home/albert/anaconda3/lib/python3.6/imp.py", line 343, in load_dynamic
return _load(spec)
ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/install_sources#common_installation_problems
for some common reasons and solutions. Include the entire stack trace
above this error message when asking for help.
The import error shows that libcublas.so.8.0 is missing, however in both cuda/lib64 and cuda-9.0/lib64 the file does exists (newer version libcublas.so.9.0). I have checked the environmental variables and all seems to be perfectly correct. Tried removing everything that is associated with nvidia from my linux including drivers, cuda toolkit, cudnn etc. and installing it all over again several times but I always end up getting the same error.
Any suggestions what could possibly go wrong or how to fix this?
Regards,
Albert