使用TensorFlow玩GTA5


小白學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版本,請立即安裝:

安裝TensorFlow

  1. 請啟動終端。

2.1 要安裝僅限CPU版本的TensorFlow,請輸入以下命令:

pip3 install --upgrade tensorflow

2.2要安裝GPU版本的TensorFlow,請輸入以下命令:

pip3 install --upgrade tensorflow-gpu

使用ancanda安裝

需要注意的是Anaconda安裝是社區支持的,沒有官方支持。

以下步驟是在Anaconda環境中安裝TensorFlow:

  1. 按照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))


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM