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

看板Python作者 (Mortis)時間5年前 (2019/12/11 15:02), 編輯推噓4(5111)
留言17則, 8人參與, 5年前最新討論串1/1
其實一開始只是想要升級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

12/11 16:02, 5年前 , 1F
創虛擬比較簡單 不用降其他東西版本
12/11 16:02, 1F

12/11 16:03, 5年前 , 2F
conda create -n tfenv python=3.7 tensorflow-gpu=2.0
12/11 16:03, 2F

12/11 16:06, 5年前 , 3F
create時也可以後面加個anaconda 把整包一起裝到虛擬環
12/11 16:06, 3F

12/11 16:06, 5年前 , 4F
12/11 16:06, 4F

12/11 16:10, 5年前 , 5F
以後更新爆掉了話 也可以整個砍掉重來
12/11 16:10, 5F

12/11 16:12, 5年前 , 6F
借問 創虛擬環境不會很混亂嗎...?
12/11 16:12, 6F

12/11 16:43, 5年前 , 7F
base才混亂吧
12/11 16:43, 7F

12/11 16:45, 5年前 , 8F
舊project 套件版本更新後就error 或套件之間不相容 不
12/11 16:45, 8F

12/11 16:45, 5年前 , 9F
用虛擬環境超容易爛掉的
12/11 16:45, 9F

12/11 16:48, 5年前 , 10F
學會創個虛擬環境可以減少很多災難XDDD
12/11 16:48, 10F

12/11 16:50, 5年前 , 11F
pip install有時候會安裝完才跟你說不相容 超可怕
12/11 16:50, 11F

12/11 17:51, 5年前 , 12F
經過這次有煩到,會乖乖裝虛擬QQ
12/11 17:51, 12F

12/11 20:40, 5年前 , 13F
官方版的話,github上fo20225有編好的whl
12/11 20:40, 13F

12/11 20:41, 5年前 , 14F
不過都是搭配最新的cuda跟cudnn,可以參考看看
12/11 20:41, 14F

12/12 01:34, 5年前 , 15F
可以試試 pipenv
12/12 01:34, 15F

12/14 12:51, 5年前 , 16F
用anaconda不創虛擬環境 excuse me WTF
12/14 12:51, 16F

12/18 15:42, 5年前 , 17F
看到樓上笑死
12/18 15:42, 17F
文章代碼(AID): #1Ty9I2kt (Python)
文章代碼(AID): #1Ty9I2kt (Python)