ubuntu16.04上的Tensorflowgpu安装问题

Question

我有一台运行Ubuntu16.04的linux机器，nvidiageforce1060。我运行的是python3.6（使用anaconda安装）和CUDA9.1

我很高兴用tensorlow用CPU编写代码，没有任何问题。然后我想安装tensorflow-gpu，这就是我遇到问题的地方。在

首次安装cudnn7.1.2成功（make并运行mnist-cudnn成功完成）

pip3 install tensorflow_gpu-1.7.0-cp36-cp36m-manylinux1_x86_64.whl

---

import tensorflow as tf

我得到以下错误

Traceback (most recent call last):
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/bony/anaconda3/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/bony/anaconda3/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/bony/anaconda3/lib/python3.6/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
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/__init__.py", line 24, in <module>
    from tensorflow.python import *  # pylint: disable=redefined-builtin
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/python/__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 74, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/bony/anaconda3/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/bony/anaconda3/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/bony/anaconda3/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/bony/anaconda3/lib/python3.6/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.

如果您能帮我解决这个问题并让我继续工作，我们将不胜感激

Answer 1

• 我在Ubuntu 16.04上也有同样的问题
• 这是我的解决方案

• 首先，在NVIDIA requirements to run TensorFlow with GPU support的官方网页部分，我们可以知道：

  • CUDA®工具箱9.0。在
  • cuDNN 7.0版。在
  • 具有CUDA计算能力的GPU卡3.0或更高版本用于从源代码构建，3.5或更高版本用于我们的二进制文件

• 所以卸载Cuda9.1。在

  • 如果不知道如何卸载，可以在Ubuntu软件中安装Synaptic Package Manager，然后搜索cuda并选择{}
  • 应用
  • 重新加载Synaptic Package Manager
  • 选择Cuda-9.0Mark for install
  • 应用

• 安装cnDNN v7.0

  • 我下载了cuDNN v7.0.5 Developer Library for Ubuntu16.04 (Deb)，因为它是Cuda-9.0的最新版本cudnv7.0

• 安装cuda-command-line-tools

  sudo apt-get install cuda-command-line-tools

  • 如果找不到cuda-command-line-tools

    sudo apt-cache search cuda-command-line-tool

    然后选择安装cuda-command-line-tool-9.0

• 按照官方文档，将其路径添加到LD_LIBRARY_path环境变量中

  export LD_LIBRARY_PATH=${LD_LIBRARY_PATH:+${LD_LIBRARY_PATH}:}/usr/local/cuda/extras/CUPTI/lib64

• 我使用的是Anaconda3，我使用conda命令创建环境包括py3.5名称py35。

• 所以，我使用source activate py35和{}。在

• 您可以选择tensorflow-gpu哪一个支持您的python版本。

• Validate your installation

  • 运行一个简短的TensorFlow程序 import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))
  • 如果系统输出Hello, TensorFlow!，那么就可以开始编写TensorFlow程序了。在

祝你好运。在

Answer 2

• 检查此github问题https://github.com/tensorflow/tensorflow/issues/15604
• 你说你已经安装了cude9.1，但它仍然显示你有cuda9.0
• 并检查您使用的tensor flow版本是否与cuda9.1兼容，因为很多评论都说tensor flow需要cuda9.0而不是cuda9.1
• 如果您使用的是Python，则可以使用以下命令：
  1] conda install -c anaconda tensorflow-gpu
  2] conda install -c aaronzs tensorflow-gpu
• 我想上面的命令还安装了cuda工具箱，并为您完成了所有必需的配置，这使您的生活更加轻松