GPU版pytorch詳細安裝過程與避坑
GPU版pytorch詳細安裝過程與避坑
前言:安裝gpu版的pytorch真的有無數多的坑，本人也是個小白，經過一頓折騰之後終於成功安裝好了。下面提供兩個安裝pytorch-gpu版比較有參考意義的兩個鏈接地址，之所以寫本篇文章，是針對以下兩篇文章不夠細緻之處做一些額外的補充說明，給大家避坑，希望大家能夠成功安裝。
說明
- 提供了兩篇安裝參考文章，本文不在大量文字詳細描述。
- 安裝GPU-pytorch版只需注意顯卡型號、CUDA、CUDNN、pytorch、torchvision、python版本型號的對應關係，即可安裝好。
1.可供參考鏈接
1.win10下pytorch-gpu安裝以及CUDA詳細安裝過程
2.Pytorch(一) Pytorch 的安裝
2.安裝步驟
- 檢查電腦是否有合適的GPU，參考上述提供的鏈接2；
- 注意顯卡的算力，和顯卡的驅動程序，以選擇合適的CUDA進行下載，點擊此處
cuda下載地址，此處需要注意的是，目前官方提供的Pytorch版本需要至少cuda8.0以上的版本，爲避免後續的麻煩，可嘗試對顯卡驅動進行相應的更新，驅動更新可利用「驅動人生」。 - 下載想用的cudnn。（注意cuda,與cudnn下載均較慢，可嘗試使用idm或者迅雷進行下載）
- 安裝pytorch與pytorchvision
以上的各步安裝過程及是否安裝成功，在上述兩個鏈接裏面都有詳細的過程，此處不再重複，僅在第3小節說明上述文章中沒寫到的細節和關鍵性問題
3.關鍵性安裝點
在前面「說明」中已經提及到，需要安裝的所以東西都需要對應，否則無法利用torch調用GPU進行運算，此處就是爲什麼很多朋友無法安裝成功的原因。
以本人安裝爲例（下載cuda時注意操作系統版本）：
- 獨立顯卡，顯卡驅動爲369.3，對應上表，所以選擇cuda8.0進行安裝，cuda大小大約有1.2G。接下來選擇對應cuda8.0的cudnn進行下載（該步不容易錯）
2. CUDA與CUDNN安裝成功後，接下來選擇合適的pytorch進行安裝，此步最關鍵，pytorch的安裝需對應你的python版本，cuda版本，以及操作系統版本。可在pytorch官網離線下載第三方庫，然後安裝。
本人電腦python爲3.7，cuda安裝的爲8.0，所以下載的pytorch版本爲0.4.1的GPU版本，詳情如下圖所示：
（注：本人顯卡配置較低，考慮各位朋友可能顯卡較好，安裝的各部分版本較高，此處只需注意如下的的torch文件命名規則即可，如圖紅框，cu80代表cuda8.0，torch-0.4.1代表torch版本，cp37代表python的安裝版本爲3.7，win_am64代表windows64位操作系統）
torchvision的安裝可直接pip,因爲其較小，參考前文提供的兩個鏈接
至此，所有安裝工作已經完畢，本文重點在第3節，其它筆者一筆帶過的安裝過程可參考文上提供的兩個鏈接，各位朋友如有啥問題，可一起交流探討，互相學習。