天天看點

Tensorflow-GPU搭建過程

Anaconda和vs需要安裝!在安裝anaconda的時候直接安裝vs即可~

值得一提的是2020年一月份還是不建議安裝python3.8以上的版本,3.6,3.7為宜。個人感覺,僅供參考。

ps:網上很多的部落格其實沒啥意思,因為每個人的電腦和相應的配置都不一樣,受之于魚不如授之于漁。下面我來說說具體方法。

先通過英偉達控制台檢視顯示卡型号驅動版本對應相應的CUDA版本(右擊桌面點選面闆即可)。一定要下載下傳對應的版本,要不然大機率失敗。安裝步驟沒啥好說的,很簡單,精簡、自定義都可以。

Tensorflow-GPU搭建過程
Tensorflow-GPU搭建過程

接下來說cudnn的配置,這也是我浪費很多時間的地方。

大緻是這樣的,網上查一查對應的版本,然後官網下載下傳安裝即可。總體來說遇到問題不要怕,按部就班來就行。主要是版本比對問題,其他的不是事兒。

繼續閱讀