windows10安裝tensorflow GPU及keras

  • 991
  • 0

這篇主要是紀錄安裝tensorflow GPU及keras的過程,還有一些注意事項。

軟體需求

  • python 3.5 或 3.6
  • anaconda (用來獨立環境,此篇使用ANACONDA NAVIGATOR來管理環境)
  • Visual C++ 2015 Redistributable Update 3
  • CUDA
  • cuDNN 

 

安裝 Microsoft Visual C++ 2015 可轉散發套件 Update 3

你可以從微軟的下載中心找到

 

安裝CUDA

請到tensorflow官網查詢支援的CUDA版本,
若版本不符,請先移除舊版CUDA再行安裝。

 

安裝cuDNN

下載cuDNN時,請選擇對應的CUDA版本下載

完成後解壓縮至CUDA目錄中即可

 

使用anaconda

我們使用ANACONDA NAVIGATOR來啟動環境,
因此不必修改ANACONDA相關的環境變數

 

建立新環境

點選Create,取一個喜歡的名字,python選擇3.5/3.6

 

啟動環境

三角按鈕,Open Terminal

 

安裝tensorflow-gpu

進入環境後
pip install tensorflow-gpu

pip install keras

就大功告成囉