cuda、torch、torchvision對應版本以及安裝
阿新 • • 發佈:2022-05-18
查詢torch與torchvision對應版本
匹配情況如下:
1.線上下載:
在pytorch官網選擇相應的歷史版本,使用conda或者pip安裝,使用官網的映象下載很慢,建議使用其他的映象源,這裡我使用的是阿里的映象速度還不錯。
pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -i https://mirrors.aliyun.com/pypi/simple/
2.離線下載
在torch網站中選擇cuda、torch、torchivsion、python的對應版本,想下載cpu版本的開頭選cpu,向下載gpu版本的選cu開頭的,注意一定要下載對應的版本,要不然會有很多錯誤!
例如針對:cu102/torch-1.7.1-cp38-cp38-linux_x86_64.whl
cu102:表示cuda版本為10.2, torch-1.7.1:表示torch版本為1.7.1 cp38:表示適用python版本為3.8 linux:表示適用於linux系統 x86_64:表示同時相容32和64位系統 下載完成後啟用進入環境,pip install 路徑\檔名.whl 安裝即可
測試
如果安裝成功,在環境中pip list即可檢視安裝的torch、torchvision、torchaudio版本
或者可以執行以下命令檢視torch的版本
python
import torch
print(torch.__version__)