小白學TensorFlow(一)
tensorflow安裝
在安裝之前,您必須選擇以下類型的TensorFlow之一來安裝:
- TensorFlow僅支持CPU支持。如果您的系統沒有NVIDIA®GPU,則必須安裝此版本。請注意,此版本的TensorFlow通常會更容易安裝(通常在5或10分鍾內),因此即使您有NVIDIA GPU,建議先安裝此版本。
- TensorFlow支持GPU。TensorFlow程序通常在GPU上的運行速度明顯高於CPU。因此,如果您的系統具有滿足以下所示先決條件的NVIDIA®GPU,並且您需要運行性能關鍵型應用程序,則應最終安裝此版本。
運行TensorFlow與GPU支持的要求
如果您使用本指南中描述的機制之一來安裝具有GPU支持的TensorFlow,則系統上必須安裝以下NVIDIA軟件:
- CUDA®工具包8.0。有關詳細信息,請參閱 NVIDIA的文檔 確保將相關的Cuda路徑名附加到%PATH% 環境變量中,如NVIDIA文檔中所述。
- 與CUDA Toolkit 8.0相關的NVIDIA驅動程序。
- cuDNN v6或v6.1。有關詳細信息,請參閱 NVIDIA的文檔。請注意,cuDNN通常安裝在與其他CUDA DLL不同的位置。記得將cuDNN DLL的目錄添加到%PATH%環境變量中。
- 具有CUDA Compute Capability 3.0或更高版本的GPU卡。有關支持的GPU卡的列表,請參閱 NVIDIA文檔。
如果您有其他版本的上述軟件包,請更改為相應的版本。特別地,cuDNN版本必須完全匹配:如果找不到,TensorFlow將不會加載cuDNN64_6.dll。要使用不同版本的cuDNN,您必須從源代碼構建。
本地安裝
如果您未安裝以下任何Python版本,請立即安裝:
-
PS: TensorFlow在Windows上支持Python 3.5.x和3.6.x
安裝TensorFlow
- 請啟動終端。
2.1 要安裝僅限CPU版本的TensorFlow,請輸入以下命令:
pip3 install --upgrade tensorflow
2.2要安裝GPU版本的TensorFlow,請輸入以下命令:
pip3 install --upgrade tensorflow-gpu
使用ancanda安裝
需要注意的是Anaconda安裝是社區支持的,沒有官方支持。
以下步驟是在Anaconda環境中安裝TensorFlow:
-
按照Anaconda下載網站上的說明 下載並安裝Anaconda。
並用命令創建一個名為tensorflow的conda環境:conda create -n tensorflow python = 3.5
通過發出以下命令激活conda環境:
activate tensorflow
3.1 要安裝僅限CPU版本的TensorFlow,請輸入以下命令:
pip install --ignore-installed --upgrade tensorflow
3.2 要安裝GPU版本的TensorFlow,請輸入以下命令(在一行中):
pip install --ignore-installed --upgrade tensorflow-gpu
查看是否安裝成功
使用ide或者shell運行
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))