本指南介紹了如何在 Ubuntu 上安裝 TensorFlow。這些指令也可能對其他 Linux 變體起作用, 但是我們只在Ubuntu 14.04 或更高版本上測試了(我們只支持) 這些指令。

一、確定要安裝的 TensorFlow

• 僅使用 CPU 支持的 TensorFlow。如果您的系統沒有 NVIDIA? GPU, 則必須安裝此版本。請註意, 此版本的 TensorFlow 通常更容易安裝 (通常在5或10分鐘), 因此, 即使您有 NVIDIA? GPU, 我們建議先安裝此版本。

• GPU 支持的TensorFlow。TensorFlow 程序在 GPU 上運行的速度通常比在 CPU 上快得多。因此, 如果您的系統有一個 NVIDIA? GPU 滿足下面所示的先決條件, 並且您需要高效運行應用程序, 則應最終安裝此版本。

2、使用 GPU 支持的 TensorFlow 的 要求

如果您使用本指南中介紹的一種機制來安裝 GPU 支持的 TensorFlow , 則必須在您的系統上安裝以下 NVIDIA 軟件:

• CUDA? Toolkit 8.0。有關詳細信息, 請參閱 NVIDIA 的文檔。請確保將相關的 Cuda 路徑名追加到 LD_LIBRARY_PATH 環境變量中, 如 NVIDIA 文檔中所述。
• 與CUDA? Toolkit 8.0 關聯的 NVIDIA 驅動程序。
• cuDNN v6.0。有關詳細信息, 請參閱 NVIDIA 的文檔。確保按照 NVIDIA 文檔中的描述，創建 CUDA_HOME 環境變量。
• 具有 CUDA 計算能力3.0 或更高的 GPU 卡。有關支持的 GPU 卡的列表, 請參見 NVIDIA 文檔。
• libcupti-dev庫, 這是 NVIDIA CUDA 配置文件的工具界面。此庫提供高級性能分析支持。要安裝此庫, 請發出以下命令:

sudo apt-get install libcupti-dev

• 從源碼中安裝 TensorFlow, 如從源碼安裝 TensorFlow。
• 至少安裝或升級到以下 NVIDIA 版本:
  • 　　CUDA toolkit 7.0或更高
  • 　　cuDNN v3 或更大
  • 　　具有 CUDA 計算能力3.0 或更高的 GPU 卡。

3、確定如何安裝 TensorFlow

• virtualenv

• "本地" pip

• Docker

• Anaconda

• 從源碼安裝, 在單獨的指南中記錄。

1）用 virtualenv安裝

請執行以下步驟以 Virtualenv 安裝 TensorFlow:

（1）通過發出以下命令之一來安裝 pip 和 virtualenv:

sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7
 sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n

（2）通過發出以下命令之一來創建 virtualenv 環境:

 virtualenv --system-site-packages targetDirectory # for Python 2.7
 virtualenv --system-site-packages -p python3 targetDirectory # for Python 3.n

其中 targetDirectory 指定 virtualenv tree的頂部。我們的指示假定 targetDirectory 是~/ tensorflow, 但您可以選擇任何目錄。

（3）通過發出以下命令之一來激活 virtualenv 環境:

source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh
 source ~/tensorflow/bin/activate.csh  # csh or tcsh

源代碼命令前面提示將更改為以下內容:

(tensorflow)$ 

（4）確保已安裝 pip ≥8.1:

(tensorflow)$ easy_install -U pip

（5）在激活的virtualenv 環境中, 發出以下命令之一以安裝 TensorFlow:

(tensorflow)$ pip install --upgrade tensorflow      # for Python 2.7
 (tensorflow)$ pip3 install --upgrade tensorflow     # for Python 3.n
 (tensorflow)$ pip install --upgrade tensorflow-gpu  # for Python 2.7 and GPU
 (tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU

如果前面的命令成功, 請跳過步驟6。如果前面的命令失敗, 請執行步驟6。

（6）（可選)如果步驟5失敗 (通常是因為您調用的 pip 版本低於 8.1), 請通過發出以下格式的命令在激活virtualenv 環境中安裝 TensorFlow:

(tensorflow)$ pip install --upgrade tfBinaryURL   # Python 2.7
 (tensorflow)$ pip3 install --upgrade tfBinaryURL  # Python 3.n 

其中 tfBinaryURL 標識了 TensorFlow Python 包的 URL。tfBinaryURL 取決於操作系統、Python 版本和 GPU 。在這裏為您的系統找到合適的 tfBinaryURL 。例如, 如果您要安裝 Linux、Python 3.4 和僅支持 CPU 的 TensorFlow, 請發出以下命令以在激活的 virtualenv 環境中安裝 TensorFlow:

(tensorflow)$ pip3 install --upgrade  https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.4.1-cp34-cp34m-linux_x86_64.whl

如果遇到安裝問題, 請參閱常見安裝問題。

$ source ~/tensorflow/bin/activate      # bash, sh, ksh, or zsh
$ source ~/tensorflow/bin/activate.csh  # csh or tcsh

當 virtualenv 環境處於活激活態時, 您可以從該 shell 運行 TensorFlow 程序。您的提示將成為以下的樣子，指示您的 tensorflow 環境處於激活狀態：

(tensorflow)$ 

使用 TensorFlow 用完後, 您可以通過調用停用函數來停用環境, 如下所示:

(tensorflow)$ deactivate 

提示符將恢復到您的默認提示 (由 PS1 環境變量定義)。

$ rm -r targetDirectory 

-------------------------------------

在 Ubuntu 上安裝 TensorFlow (官方文檔的翻譯)