CrownTail的部落格

今天參考網路的程式碼測試CNN

總共有242062個參數

準備要fit model時發現速度極慢

沒有耐性等model train完

上網查了一下，原來tensorflow還有分成CPU版本與GPU版本

如果要安裝GPU版本，還需要安裝 nvida CUDA &  與下載cnDNN

像這樣 * 卡在第一個epochs

CPU滿載；但兩顆GPU都沒用到

接著來安裝 CUDA / cnDNN

檢查tensorflow 支援幾版的CUDA /cuDNN

CUDA 9.0 與 cuDNN 7.0

下載安裝 CUDA，這部份很簡單，一直按下一步下一步下一步下一步就好了 

(((眼尖的網友，應該就知道我犯了個嚴重的錯誤阿

下載cuDNN，有很多選擇，找出對應的版本即可，v7.0 for windows

解壓縮後將bin\DLL位置加入系統環境變數

建立CONDA 工作環境 : conda create --name trensorflow-gpu python=3.6 anaconda nb_conda

中間卡在Verifying transaction 卡了一個晚上，後來胡亂按了enter 之類的才又繼續run

跑完後如下

試run，jupyter notebook 報錯，找不到cudart64_90.dll

因為安裝成cuda 9.2，改下載9.0版本重新安裝

重新安裝CUDA9.0 就OK!!