我正在尝试安装带有 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
你能帮我解决这个问题吗
这几乎总是你的缺失路径LD_LIBRARY_PATH
. Find 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 是典型的,但不是必需的)。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)