前往
大廳
主題

電腦重灌&tensorflow-gpu+CUDA安裝心得

LMz | 2020-12-04 13:46:09 | 巴幣 0 | 人氣 254

因為電腦有各種令人煩躁的問題所以選擇重灌
為此還特地去場外發了一篇文問
灌下來雖然很簡單
但灌完之後才有一堆問題又重灌一次

我直接從結論講
就是重灌的過程中不要連網路
因為win10會自動上網幫忙抓驅動
像我的顯示卡是gtx1060,win10安裝過程自動幫我抓了427版的驅動,大概是2019年末的版本
問題是400以後的版本都是給20系以上的顯示卡用,支援光線追蹤啥的
1060根本不需要,10系顯卡停在399.24版就好
加上我要做機器學習,顯示卡版本不建議太新(看要用第幾版的tensorflow,我用1.10,如果要2.0以上可能就要更新)
所以重灌了一次win10,這次不連網,另外準備了399.24的驅動程式安裝軟體
灌完之後裝驅動就可以插網路了

接著安裝CUDA+cudnn
要用tensorflow-gpu的人一定要有的預備知識
那就是tensorflow / CUDA / cudnn / 顯示卡驅動版本
四個東西 缺一不可
而且是相互對應,每個版本對應其他三個版本
這東西要去查,不是隨便抓來裝就好
我自己是tensorflow-gpu=1.10 / keras = 2.2.0 / CUDA = 9.2(新版) / cudnn = 7.2.1.38 / 驅動版本=399.24
五個數字錯一個程式就會噴error,有夠搞剛
如果想要裝不同數字的麻煩自己去查對應的版本
然後就可以盡情地操gpu了
MNIST60000筆data跑一次epoch不用5秒,真爽

大概就這樣

創作回應

更多創作