出现问题:
ImportError: Traceback (most recent call last):
File "/storage/xuminghong/.local/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/storage/xuminghong/.local/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/storage/xuminghong/.local/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 "/usr/local/lib/python3.6/imp.py", line 242, in load_module
return load_dynamic(name, filename, file)
File "/usr/local/lib/python3.6/imp.py", line 342, in load_dynamic
return _load(spec)
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
原因:
tensorflew(1.7.0)版本与CUDA(8.0.44)版本不匹配
解决:
将tensorflew版本降低为1.4.0
pip install --upgrade tensorflow-gpu==1.4
或者,安装9.0版本的CUDA
sudo apt-get install cuda-9-0
附:版本匹配信息