Question

我正在尝试使用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

你能帮我解决这个问题吗？

Answer 1

这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是典型的，但不是必需的。）

Answer 2

我已安装cuda-9.0，并且LD_LIBRARY_PATH指向正确的位置。原来，文件夹/usr/local/cuda-9.0的权限是错误的。我什至无法cd到该文件夹。我将cuda-9.0的所有者从root更改为my-user-name，然后python能够找到丢失的库。

Answer 3

要轻松找到libcublas文件，请运行sudo find / -iname 'libcublas*'。然后将包含该文件的文件夹的路径添加到LD_LIBRARY_PATH中。

ImportError：libcublas.so.9.0：无法打开共享对象文件：在Ubuntu 16.04.03上安装张量流时没有这样的文件或目录

3 个答案: