[心得] Win10 TensorFlow 2.0 GPU 升級中的坑

作者: brad0315 (老默)   2019-12-11 15:02:23
其實一開始只是想要升級Spyder到4.0的版本,這樣就有內建暗黑主題可以用了。
但所有更新都是災難的開始......
廢話不多說,直接上我最後解決一切問題的步驟:
Step 1、打開Anaconda Prompt 進到你想更改的環境裡,懶人如我沒有建虛擬環境。
Step 2、更新conda :conda update conda
Step 3、更新Anaconda :conda update anaconda
Step 4、更新Spyder :conda update spyder
基本上到這邊,Spyder就會更新成新版有內建暗黑主題的,但是可能會發現自己的
TensorFlow變成跑CPU版,速度差了十幾倍。
Step 5、打開命令提示字元(cmd)
Step 6、安裝TensorFlow2.0:pip install tensorflow-gpu
Step 7、安裝 CUDA以及cuDNN: conda install cudnn=7.6 cudatoolkit=10.0
一定要這個版本,因為目前TensorFlow2.0還不支援最新CUDA10.1。
現在居然可以直接用conda安裝,真的太感動,以前還要去官網載。
Step 8、修改gast版本:pip install gast==0.2.2
因為gast最新版本有Bug,所以請安裝這個特定版本,不然會有Error,雖然可以無視。
也寫成網誌了,裡面有提供簡單的測試程式碼
有興趣的可以去看一下,無廣告、音樂:
https://mortis.tech/2019/12/tutorial/921/
作者: roccqqck (ccqq)   2019-12-11 16:02:00
創虛擬比較簡單 不用降其他東西版本conda create -n tfenv python=3.7 tensorflow-gpu=2.0create時也可以後面加個anaconda 把整包一起裝到虛擬環境以後更新爆掉了話 也可以整個砍掉重來
作者: wargods8402 (維尼X維尼)   2019-12-11 16:12:00
借問 創虛擬環境不會很混亂嗎...?
作者: roccqqck (ccqq)   2019-12-11 16:43:00
base才混亂吧舊project 套件版本更新後就error 或套件之間不相容 不用虛擬環境超容易爛掉的
作者: TuCH (謬客)   2019-12-11 16:48:00
學會創個虛擬環境可以減少很多災難XDDD
作者: roccqqck (ccqq)   2019-12-11 16:50:00
pip install有時候會安裝完才跟你說不相容 超可怕
作者: brad0315 (老默)   2019-12-11 17:51:00
經過這次有煩到,會乖乖裝虛擬QQ
作者: lionell   2019-12-11 20:40:00
官方版的話,github上fo20225有編好的whl不過都是搭配最新的cuda跟cudnn,可以參考看看
作者: s860134 (s860134)   2019-12-12 01:34:00
可以試試 pipenv
作者: yoyololicon (蘿莉大好)   2019-12-14 12:51:00
用anaconda不創虛擬環境 excuse me WTF
作者: ruokcnn (Dean)   2019-12-18 15:42:00
看到樓上笑死

Links booklink

Contact Us: admin [ a t ] ucptt.com