安装GPU版本torch
下载对应版本pytorch和torchvision
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
python、pytorch和torchvision部分版本对应。注意:自己电脑的cuda版本也应该匹配
python | pytorch | torchvision |
---|---|---|
>=3.8, <=3.12 | 2.3 | 0.18 |
>=3.8, <=3.11 | 2.2 | 0.17 |
>=3.8, <=3.11 | 2.1 | 0.16 |
>=3.8, <=3.11 | 2.0.0 | 0.15.1 |
>=3.7.2, <=3.10 |