這篇主要是紀錄安裝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
就大功告成囉