在 MX150 上安装 tensorflow-gpu 时出错答案

【问题标题】：Error while Installing tensorflow-gpu on MX150在 MX150 上安装 tensorflow-gpu 时出错
【发布时间】：2020-02-03 07:26:07
【问题描述】：

我正在尝试在运行 MX150 的笔记本电脑上安装 tensorflow gpu。我更新了我的 NVIDIA 驱动程序，安装了 CUDA，安装了 CUDnn。但仍然无法导入 Tensorflow。出现以下错误

>>> import tensorflow as tf
Traceback (most recent call last):
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/rohan/anaconda3/envs/tensorflow_env/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 "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/__init__.py", line 24, in <module>
    from tensorflow.python import *
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/python/__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 72, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/rohan/anaconda3/envs/tensorflow_env/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 "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/rohan/anaconda3/envs/tensorflow_env/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

【问题讨论】：

标签： linux python-3.x tensorflow caffe

【解决方案1】：

您的 python 版本和 CUDA 版本之间似乎不匹配。请通过以下方式检查您的版本：

python --version

和

nvcc --version

你的操作系统是什么？您正在尝试运行哪个版本的 Tensorflow？我建议在您拥有当前版本的 Python 和 CUDA 后卸载 tensorflow 并重新安装它。你可以关注Tesnoeflow with gpu documentation。

还要检查与您的问题类似的问题 libcublas.so.8.0 error with tensorflow

【讨论】：

python version-3.6.5 cuda version- 10.0, tensorflow-gpu-1.4.0 我在 ubuntu 18.04 上运行它，我可以通过终端获取 tensorflow 的版本但无法导入 tensorflow。
根据Install GPU version of tensorflow - 我建议您卸载 CUDA 并重新安装 CUDA 8.0 和 cuDNN 6.0 。链接中还提供了如何执行此操作的说明。
我无法继续，因为我的 ubuntu 版本是 18.04，而 cuda 8 可用于 ubuntu 16.04 和 14.04
所以根据表格here你需要TF版本> 1.6.0

【解决方案2】：

您需要带有 cuda 工具包 10.0 的 python 版本 3.7 我也有同样的错误，并通过在系统高级设置中设置 cuda 路径的环境变量来解决它。

如果您安装了两个或多个 cuda 版本，请确保从环境变量中删除其余部分。

使用以下代码检查 gpu 是否在 jupyter notebook 中被使用。

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

【讨论】：