[譯] 系列教程:選擇準備安裝的 TensorFlow 型別

LeviDing發表於2017-12-11

本文出自 掘金翻譯計劃 正在組織翻譯的 TensorFlow 官方文件。如果您有興趣,歡迎 申請成為譯者,學習完譯者教程後,參與到文章和文件的翻譯和及對當中。我們也正在招募 TensorFlow 譯者,歡迎積極參加。


本指南說明了如何在 Windows 系統上安裝 TensorFlow。

選擇準備安裝的 TensorFlow 型別

從以下選項中選擇您需要安裝的 TensorFlow 型別:

  • 僅支援 CPU 的 TensorFlow。 如果系統無 NVIDIA® GPU,則必須安裝該版本。需要說明的是,該版本的 TensorFlow 相比另一版本更容易安裝(通常 5 到 10 分鐘即可完成安裝),因此即使系統包含 NVIDIA GPU,我們仍然推薦您優先安裝該版本。

  • 支援 GPU 的 TensorFlow。 一般而言,TensorFlow 程式在 GPU 上的執行速度要明顯高於在 CPU 上的。因此,如果您的系統含符合以下先決條件的 NVIDIA ® GPU,且需要執行效能關鍵型應用程式,那麼您最終需要安裝此版本的 TensorFlow。

執行支援 GPU 版本 TensorFlow 的要求

若使用本指南中介紹的任一方式來安裝支援 GPU 的 TensorFlow,那麼您必須在系統中安裝如下 NVIDIA 軟體:

  • CUDA® Toolkit 8.0。詳細說明請檢視 NVIDIA 官方文件。請確保您已按照 NVIDIA 官方文件描述將相關的 Cuda 路徑名稱新增到 %PATH% 環境變數中。

  • 與 CUDA Toolkit 8.0 相關的 NVIDIA 驅動。

  • cuDNN v6.1版本。詳細說明請檢視 NVIDIA 官方文件。需要注意的是,一般而言,cuDNN 的安裝地址和其他 CUDA DLL 是不同的。同時,請確保將安裝 cuDNN DLL 的目錄新增到 %PATH% 環境變數中。。

  • 帶有 CUDA Compute Capability 3.0 或更高版本的 GPU 卡。請在 [NVIDIA 官方文件](http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/) 中查詢具備條件的 GPU 清單。

如果您使用的版本與上述要求不一致,請更新為規定的版本。特別說明的是,cuDNN 的版本必須與要求的一致:如果無法找到 cuDNN64_6.dll,那麼 TensorFlow 將無法載入。如果您想使用其他版本的 cuDNN,您需要從原始碼開始重新編譯。

選擇安裝 TensorFlow 的方式

您需要選擇安裝 TensorFlow 的方式。當前的可選方式如下:

  • 原生的 pip
  • 使用 Anaconda

原生 pip 無需安裝虛擬環境,可以直接在系統安裝 TensorFlow。由於原生的一個 pip 安裝應用並沒有被隔離在一個獨立的應用中,使用 pip 安裝方法可能會影響到系統裡其他基於 Python 的安裝。但是,如果您瞭解您系統裡的 pip 和 Python 環境,那麼使用原生 pip 安裝僅僅只需要一條命令就夠了。而且,如果您使用原生的 pip 安裝方法,那麼使用者可以從系統的任何路徑去執行 TensorFlow 程式。

在 Anaconda 中,你可以使用 conda 去建立一個虛擬環境(virtural environment)。但是,如果是使用 Anaconda 方式,我們依然推薦使用 pip 安裝命令來安裝 TensorFlow,而不是 conda 安裝命令。

注意: conda 包是由社群提供的,而不是官方。也就是說,TensorFlow 團隊並不會測試也不會維護 conda 包。使用 conda 包需要您自己承擔風險。

使用原生pip安裝

如果您的機器上沒有安裝以下版本的Python,請立刻安裝:

在 Windows 上,TensorFlow 支援 Python3.5.x 版本和 Python 3.6.x 版本。需要注意的是, Python 3 使用的是 pip3 包管理, 這也是您用來安裝 TensorFlow 的程式。 開啟一個終端,開始安裝 TensorFlow。然後在終端上執行正確的 pip3 安裝命令。 安裝僅支援 CPU 版本的 TensorFlow,請輸入下面的命令: C:\> pip3 install --upgrade tensorflow

安裝 GPU 版本的 TensorFlow,請輸入下面的命令: C:\> pip3 install --upgrade tensorflow-gpu

使用 Anaconda 進行安裝

Anaconda 的安裝包是由社群提供,非官方提供的。 在 Anaconda 的環境下,按照以下步驟進行 TensorFlow 的安裝: 1.根據網頁 [Anaconda 下載站點](https://www.anaconda.com/download/)說明下載並安裝 Anaconda。 2.請通過使用以下命令來建立一個名為 tensorflow 的 conda 環境: C:\> conda create -n tensorflow pip python=3.5 3.通過輸入以下命令來啟用一個 conda 環境: C:\> activate tensorflow (tensorflow)C:\> # Your prompt should change 4.在 conda 環境裡輸入正確的命令來安裝 TensorFlow。 安裝僅支援 CPU 版本的 TensorFlow,請輸入下面的命令: (tensorflow)C:\> pip install --ignore-installed --upgrade tensorflow 如果是安裝 GPU 版本的 TensorFlow,請輸入下面的命令: (tensorflow)C:\> pip install --ignore-installed --upgrade tensorflow-gpu

安裝驗證

開啟一個終端。 如果您採用 Anaconda 方式安裝,則進入 Anaconda 環境。 採用下列方式從你的 shell 啟用 python: $python 在 python 互動 shell 中輸入下列程式碼:

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
複製程式碼

如果系統的輸出如下所示,那就說明您可以開始在上面撰寫 TensorFlow 的程式了:

Hello, TensorFlow!

如果您剛剛接觸 TensorFlow,請參考文件TensorFlow 入門指南

如果系統輸出了一個錯誤資訊而不是一個打招呼提示,請檢視常見安裝問題

這裡也有一個關於 Windows 上 TensorFlow 安裝很有用的文件

常見安裝問題

我們使用 Stack Overflow 來記錄 TensorFlow 的安裝問題和修正方法。下表中包含有一些常見安裝問題在 Stack Overflow 上的回答連結。如果您遇到的錯誤訊息或安裝問題不在下表中,請在 Stack Overflow 上搜尋它的答案。如果 Stack Overflow 上並沒有顯示這個錯誤訊息或者安裝問題的答案,請在 Stack Overflow 上提一個關於這個錯誤訊息或者安裝問題的新問題,並給這個問題指定一個 tensorflow 的標籤。

Stack Overflow 連結 錯誤訊息
41007279 [...\stream_executor\dso_loader.cc] Couldn't open CUDA library nvcuda.dll
41007279 [...\stream_executor\cuda\cuda_dnn.cc] Unable to load cuDNN DSO
42006320 ImportError: Traceback (most recent call last): File "...\tensorflow\core\framework\graph_pb2.py", line 6, in
from google.protobuf import descriptor as _descriptor
ImportError: cannot import name 'descriptor'
42011070 No module named "pywrap_tensorflow"
42217532 OpKernel ('op: "BestSplits" device_type: "CPU"') for unknown op: BestSplits
43134753 The TensorFlow library wasn't compiled to use SSE instructions

掘金翻譯計劃 是一個翻譯優質網際網路技術文章的社群,文章來源為 掘金 上的英文分享文章。內容覆蓋 AndroidiOS前端後端區塊鏈產品設計人工智慧等領域,想要檢視更多優質譯文請持續關注 掘金翻譯計劃官方微博知乎專欄

相關文章