win10安裝帶CUDA的Pytorch看這篇就夠了

大部分的方法,都是讓你去pytorch官網的這個連結,用選擇器生成一個pip/conda命令。你點進去可能就傻眼了:

win10安裝帶CUDA的Pytorch看這篇就夠了

這裡cuda就特麼兩個版本可供咱選擇!

(當然如果你要裝的是CPU版本,看到這裡就可以叉掉本篇了)

真相是:

最好就別用那個生成的命令讓pip自己去下載安裝,

而是你找好wheel檔案、下載到本地後再pip/conda安裝。

而安裝。whl檔案前可能還特麼必須得先重新命名一下。

——————————————————————————分割線——————————————————————————

安裝環境:NVIDIA獨顯,win10

1,安裝CUDA

1。1 檢視自己顯示卡最高支援的CUDA版本

右擊桌面,進入NVIDIA控制面板

win10安裝帶CUDA的Pytorch看這篇就夠了

系統資訊-元件,可以看到我的顯示卡最高支援CUDA10。1。我可以裝CUDA9。0、9。2,但不能裝10。2

win10安裝帶CUDA的Pytorch看這篇就夠了

1。2 檢視pytorch支援的cuda版本

坑來了!但是你下載了的cuda,並不一定有對應的pytorch版本!

這是pytorch所有wheel的下載頁面:

wheel檔名中的cu102表示cuda10。2,以此類推。

比如我另一臺電腦最高支援cuda11。2,但裝完cuda11。2然後去了pytorch的wheel下載頁面,發現cu111有cu113就是沒有cu112,就傻眼了,只得解除安裝重灌cuda11。1

win10安裝帶CUDA的Pytorch看這篇就夠了

1。3 下載CUDA和cuDnn安裝包

我們要下載的是這兩個東西:

win10安裝帶CUDA的Pytorch看這篇就夠了

大陸訪問NVIDIA網站賊慢,甚至可能打不開,要命的是還

必須登入才能下載!

所以最好有梯子

1。3。1 下載CUDA

CUDA下載地址:

選擇自己要的版本下載即可

win10安裝帶CUDA的Pytorch看這篇就夠了

注意:

CUDA一定要選擇“exe[local]”!!!線上安裝包是巨坑,2G多的安裝檔案下載個半天,一旦安裝結束就刪除,如果安裝失敗了,那你重新安裝時還得再慢慢下載一遍。

1。3。2 下載cudnn

根據自己的CUDA版本下載對應的cudnn

1。4 安裝CUDA

1。4。1 選自定義

win10安裝帶CUDA的Pytorch看這篇就夠了

1.4.2 根據需要勾選安裝的元件,建議第一次安裝時只安裝CUDA核心元件,其他的後面有需要再裝

win10安裝帶CUDA的Pytorch看這篇就夠了

尤其最好先別裝Visual Studio Integration。因為如果安裝失敗基本都是它導致的。

win10安裝帶CUDA的Pytorch看這篇就夠了

1。4。3 單擊“下一步”安裝。

1。5 安裝cudnn

1。5。1 解壓下載的cudnn壓縮包

win10安裝帶CUDA的Pytorch看這篇就夠了

1.5.2 去CUDA安裝目錄貼上三個資料夾,如果提示重名則覆蓋。CUDA預設安裝目錄如下:

win10安裝帶CUDA的Pytorch看這篇就夠了

2 安裝pytorch

2。1 下載wheel檔案

上面那個網址()是給出了所有版本的pytorch。

其實還可以訪問這個:

https://download。pytorch。org/whl/cu+你的cuda版本號

例如cuda11。3,那就是訪問https://download。pytorch。org/whl/cu113

我是cuda10。1,那就是訪問https://download。pytorch。org/whl/cu101

2。1。1 單擊你要安裝的庫,要裝pytorch就單擊torch:

win10安裝帶CUDA的Pytorch看這篇就夠了

2.1.2 根據自己的環境找whl檔案,:

這裡需要

注意

1,我們是已知CUDA版本來找pytorch的whl檔案,別找到CPU版去了

我下載的這個檔名:torch-1。8。1+cu101-cp38-cp38-win_amd64。whl

意思是pytorch版本為1。8。1,CUDA版本為10。1,python版本為3。8,windows64位

2,如果是安裝torchvision,它的版本要和torch版本匹配!匹配關係如下表

win10安裝帶CUDA的Pytorch看這篇就夠了

該表來自

2。1。3 下載該whl檔案,並pip安裝

這裡是個坑,這些whl可能直接就能安裝,也可能必須改名才能安裝,也是服了。

如果直接pip install ****。whl失敗,就需要把末尾的“win_amd64”改成“win32”:

win10安裝帶CUDA的Pytorch看這篇就夠了

既然是win10,就別隻知道cmd命令行了。

右擊在whl檔案所在目錄的空白處 ,開啟Powershell:

win10安裝帶CUDA的Pytorch看這篇就夠了

pip install 該whl,成功:

win10安裝帶CUDA的Pytorch看這篇就夠了