我正在尝试安装支持cuda和gpu的tensorflow。当我尝试导入它时,我得到以下错误:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/srivishnu/.local/lib/python2.7/site-packages/tensorflow/__init__.py", line 24, in <module>
from tensorflow.python import *
File "/home/srivishnu/.local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
File "/home/srivishnu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 74, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "/home/srivishnu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/home/srivishnu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/home/srivishnu/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
此错误发生时,我键入此命令:
import tensorflow as tf
你能帮我解决这个问题吗
3条答案
按热度按时间6qqygrtg1#
这几乎总是
LD_LIBRARY_PATH
中缺少的路径。在系统上查找libcublas.so.9.0
(从/usr/local
下开始查找)。如果找不到,请安装CUDA 9.0 Toolkit(注意,对于TF 1.5,您需要9.0而不是9.1,这是一个常见的问题,至少就我自己犯的错误而言是常见的)。
https://developer.nvidia.com/cuda-90-download-archive
如果有,请更新
LD_LIBRARY_PATH
,使其指向相应的lib
目录。如果您已经完成了其中的任何一个,并且现在在cudnn相关库中得到了类似的查找错误,那么请对CUDNN库重复该过程。
https://developer.nvidia.com/cudnn
我认为最新的版本是有效的。Tensorflow依赖于CUDA工具包和CuDNN库扩展。
请注意,您也可以在用户空间中安装所有这些(sudo是典型的,但不是必需的)。
2exbekwf2#
要轻松查找
libcublas
文件,请运行或
然后将包含该文件的文件夹的路径添加到
LD_LIBRARY_PATH
中。wvt8vs2t3#
我安装了
cuda-9.0
,LD_LIBRARY_PATH
指向了正确的位置。原来文件夹/usr/local/cuda-9.0
的权限是错误的。我甚至不能cd
到文件夹。我将cuda-9.0
的所有者从root
更改为my-user-name
,之后python能够找到丢失的库。