今天用了下面的命令:
pip install tensorflow-gpu
然后运行程序的时候出现了下面的错误信息:
......low/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File "/home/idc/anaconda3/envs/imagecaption/lib/python3.5/imp.py", line 243, in load_module
return load_dynamic(name, filename, file)
File "/home/idc/anaconda3/envs/imagecaption/lib/python3.5/imp.py", line 343, in load_dynamic
return _load(spec)
ImportError: libcublas.so.9.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.
这是因为我装了tensorflow-gpu 1.5版本,而我用的是cuda 8.0和cudnn6.0,1.5版本要求cuda 9.0,我的做法就是回滚:
pip install tensorflow-gpu==1.4
然后运行就没有报错了。
参考文献
[1].tensorflow暂未兼容cuda9.0, 安装cuda8.0.http://blog.csdn.net/line290/article/details/78534828