大部分的方法,都是讓你去pytorch官網的這個連結,用選擇器生成一個pip/conda命令。你點進去可能就傻眼了:
這裡cuda就特麼兩個版本可供咱選擇!
(當然如果你要裝的是CPU版本,看到這裡就可以叉掉本篇了)
真相是:
最好就別用那個生成的命令讓pip自己去下載安裝,
而是你找好wheel檔案、下載到本地後再pip/conda安裝。
而安裝。whl檔案前可能還特麼必須得先重新命名一下。
——————————————————————————分割線——————————————————————————
安裝環境:NVIDIA獨顯,win10
1,安裝CUDA
1。1 檢視自己顯示卡最高支援的CUDA版本
右擊桌面,進入NVIDIA控制面板
系統資訊-元件,可以看到我的顯示卡最高支援CUDA10。1。我可以裝CUDA9。0、9。2,但不能裝10。2
1。2 檢視pytorch支援的cuda版本
坑來了!但是你下載了的cuda,並不一定有對應的pytorch版本!
這是pytorch所有wheel的下載頁面:
wheel檔名中的cu102表示cuda10。2,以此類推。
比如我另一臺電腦最高支援cuda11。2,但裝完cuda11。2然後去了pytorch的wheel下載頁面,發現cu111有cu113就是沒有cu112,就傻眼了,只得解除安裝重灌cuda11。1
1。3 下載CUDA和cuDnn安裝包
我們要下載的是這兩個東西:
大陸訪問NVIDIA網站賊慢,甚至可能打不開,要命的是還
必須登入才能下載!
所以最好有梯子
1。3。1 下載CUDA
CUDA下載地址:
選擇自己要的版本下載即可
注意:
CUDA一定要選擇“exe[local]”!!!線上安裝包是巨坑,2G多的安裝檔案下載個半天,一旦安裝結束就刪除,如果安裝失敗了,那你重新安裝時還得再慢慢下載一遍。
1。3。2 下載cudnn
根據自己的CUDA版本下載對應的cudnn
1。4 安裝CUDA
1。4。1 選自定義
1.4.2 根據需要勾選安裝的元件,建議第一次安裝時只安裝CUDA核心元件,其他的後面有需要再裝
尤其最好先別裝Visual Studio Integration。因為如果安裝失敗基本都是它導致的。
1。4。3 單擊“下一步”安裝。
1。5 安裝cudnn
1。5。1 解壓下載的cudnn壓縮包
1.5.2 去CUDA安裝目錄貼上三個資料夾,如果提示重名則覆蓋。CUDA預設安裝目錄如下:
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:
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版本匹配!匹配關係如下表
(
該表來自
)
2。1。3 下載該whl檔案,並pip安裝
這裡是個坑,這些whl可能直接就能安裝,也可能必須改名才能安裝,也是服了。
如果直接pip install ****。whl失敗,就需要把末尾的“win_amd64”改成“win32”:
既然是win10,就別隻知道cmd命令行了。
右擊在whl檔案所在目錄的空白處 ,開啟Powershell:
pip install 該whl,成功: