ubuntu16.04上的Tensorflow-gpu安裝問題

Question

我有一台運行Ubuntu16.04，NVIDIA GEFORCE1060的Linux機器。 我正在運行python3.6（使用anaconda安裝）和CUDA9.1

我很高興在Tensorlow中使用CPU進行編碼，沒有任何問題。 然后我想安裝tensorflow-gpu ，這就是我遇到的問題。

首次成功安裝cudnn7.1.2（成功完成make and run 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.

任何幫助解決此問題並讓我繼續使用tensorflow-gpu將不勝感激

Answer 1

• 檢查此github問題https://github.com/tensorflow/tensorflow/issues/15604
• 您說您已經安裝了cude9.1，但仍然表明您已經安裝了cuda9.0
• 並檢查您使用的張量流版本是否與cuda9.1兼容，因為許多評論都說張量流期望cuda9.0而不是cuda9.1
• 如果使用anaconda，則可以使用以下命令：
  1] conda install -c anaconda tensorflow-gpu
  2] conda install -c aaronzs tensorflow-gpu
• 我認為上述命令還可以安裝cuda-toolkit並為您完成所有必需的配置，從而使您的生活更輕松

Answer 2

• 我在Ubuntu 16.04上遇到了同樣的問題
• 這是我的解決方案

• 首先，在NVIDIA要求的官方網頁上， 要運行具有GPU支持的TensorFlow，我們可以知道：

  • CUDA®工具包9.0。
  • cuDNN v7.0。
  • 具有CUDA Compute Capability 3.0或更高版本的GPU卡，可用於從源代碼構建，而具有3.5或更高版本的二進制文件

• 因此，請卸載Cuda9.1。

  • 如果您不知道如何卸載，則可以在Ubuntu軟件中安裝Synaptic Package Manager ，然后搜索cuda並選擇“ Mark for complete removal
  • 應用
  • 重新加載Synaptic Package Manager
  • 選擇Cuda-9.0 Mark for install
  • 應用

• 安裝cnDNN v7.0

  • 我已經下載了cuDNN v7.0.5 Developer Library for Ubuntu16.04 (Deb) ，因為它是Cuda-9.0的最新版本cuDNN v7.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

    然后選擇install 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並使用pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.7.0-cp35-cp35m-linux_x86_64.whl source activate py35 pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.7.0-cp35-cp35m-linux_x86_64.whl 。

• 您可以選擇tensorflow-gpu ，其中一個支持您的python版本。

• 驗證安裝

  • 運行一個簡短的TensorFlow程序， import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))
  • 如果系統輸出Hello, TensorFlow! ，那么您就可以開始編寫TensorFlow程序了。

祝你好運。