pytorch的官方下载地址是:
https://download.pytorch.org/whl/torch_stable.html
需要安装的是torch、torchvision。版本分为支持cpu和gpu的,而且很多,需要注意寻找。
安装命令是
pip install torch-1.10.0+cu113-cp39-cp39-win_amd64.whl
pip install torchvision-0.12.0+cpu-cp39-cp39-win_amd64.whl
需要注意的是torch和torchvision的版本是不一样的。这是第一个坑。由于torch包很大,建议先安装torchvisio