1. 程式人生 > >Win10 GPU 環境下的 Tensorflow 配置

Win10 GPU 環境下的 Tensorflow 配置

安裝須知:

TensorFlow 有cpu和 gpu兩個版本。gpu版本需要英偉達CUDA 和 cuDNN 的支援,cpu版本不需要。這裡我裝的是gpu版,

Win10_X64。

準備工作(共四步):

Step1: 確認你的顯示卡支援 CUDA,運算能力符合你要執行的程式碼的運算量(一般在論文的  實驗中會說明,請自行檢視)。

  • 顯示介面卡(顯示卡)檢視方式:在cmd中輸入”start devmgmt.msc”。

  • 我的顯示卡GPU: NVIDA GeForce GTX 750  

  • GPU運算能力查詢:查詢頁面

  • 我的顯示卡運算能力: 5.0

Step2: 確認你安裝了VS2015或者2013或者2010,我用的2013。

Step3: python 3.5(確認你的Python版本是3.5 64位,不建議使用更高的Python版本。關於python建議安裝Anaconda,因為這個集成了很多科學計算所必需的庫,能夠避免很多依賴問題,安裝也特別方便。) 

  • Anaconda 官網最近版本:下載頁面

  • Anaconda 官網所有版本:下載頁面

  • 版本選擇:Anaconda3-4.2.0-Windows-x86_64.exe (該版本的Anaconda自帶的Python版本就是3.5)

 Step4:  pip 9.0.1(確認pip版本至少為8.1,用pip -V 檢視當前 pip 版本,用“python –m3 pip install -U pip”升級pip 。)

做完以上準備工作,你可以稍作休息,然後檢視第二篇部落格,進行後續工作。