python安装对应版本的cuda和torch以及torchvision

Pytorch+cuda+torchvision的安装命令

当使用pip命令时，windows环境下，可以仿照下面的命令对应上面的版本进行下载！

pip install torch1.12.0+rocm5.1.1 torchvision0.13.0+rocm5.1.1 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

也可以到pytorch官网链接 https://pytorch.org/get-started/previous-versions/

鼠标滚轮下滑 ↓ 找到类似下面的下载方式！

复制到终端cmd下，复制命令即可！

CUDA 11.6

pip install torch1.12.0+cu116 torchvision0.13.0+cu116 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 11.3 ****我用到的

pip install torch1.12.0+cu113 torchvision0.13.0+cu113 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 10.2

pip install torch1.12.0+cu102 torchvision0.13.0+cu102 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.12.0+cpu torchvision0.13.0+cpu torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

完成