作者brad0315 (Mortis)
看板Python
標題[心得] Win10 TensorFlow 2.0 GPU 升級中的坑
時間Wed Dec 11 15:02:23 2019
其實一開始只是想要升級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/
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 140.113.76.101 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1576047746.A.BB7.html
→ roccqqck: 創虛擬比較簡單 不用降其他東西版本 12/11 16:02
→ roccqqck: conda create -n tfenv python=3.7 tensorflow-gpu=2.0 12/11 16:03
→ roccqqck: create時也可以後面加個anaconda 把整包一起裝到虛擬環 12/11 16:06
→ roccqqck: 境 12/11 16:06
推 roccqqck: 以後更新爆掉了話 也可以整個砍掉重來 12/11 16:10
推 wargods8402: 借問 創虛擬環境不會很混亂嗎...? 12/11 16:12
→ roccqqck: base才混亂吧 12/11 16:43
→ roccqqck: 舊project 套件版本更新後就error 或套件之間不相容 不 12/11 16:45
→ roccqqck: 用虛擬環境超容易爛掉的 12/11 16:45
推 TuCH: 學會創個虛擬環境可以減少很多災難XDDD 12/11 16:48
→ roccqqck: pip install有時候會安裝完才跟你說不相容 超可怕 12/11 16:50
→ brad0315: 經過這次有煩到,會乖乖裝虛擬QQ 12/11 17:51
→ lionell: 官方版的話,github上fo20225有編好的whl 12/11 20:40
→ lionell: 不過都是搭配最新的cuda跟cudnn,可以參考看看 12/11 20:41
推 s860134: 可以試試 pipenv 12/12 01:34
噓 yoyololicon: 用anaconda不創虛擬環境 excuse me WTF 12/14 12:51
推 ruokcnn: 看到樓上笑死 12/18 15:42