Windows10

• 以 cuda 10.2 版本为例

torch 1.9.0

# python3.7
pip install https://download.pytorch.org/whl/cu102/torch-1.9.0%2Bcu102-cp37-cp37m-win_amd64.whl

# python3.8
pip install https://download.pytorch.org/whl/cu102/torch-1.9.0%2Bcu102-cp38-cp38-win_amd64.whl

torchvision 0.10.0

# python3.7
pip install https://download.pytorch.org/whl/cu102/torchvision-0.10.0%2Bcu102-cp37-cp37m-win_amd64.whl

# python3.8
pip install https://download.pytorch.org/whl/cu102/torchvision-0.10.0%2Bcu102-cp38-cp38-win_amd64.whl

Ubuntu

其他版本可以参考此处