1. 程式人生 > 其它 >cuda、torch、torchvision對應版本以及安裝

cuda、torch、torchvision對應版本以及安裝

查詢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__)