博客

Pytorch+cuda+torchvision的安装命令

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

pip install torch

1.12.0+rocm5.1.1 torchvision

0.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 torch

1.12.0+cu116 torchvision

0.13.0+cu116 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 11.3 ****我用到的

pip install torch

1.12.0+cu113 torchvision

0.13.0+cu113 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 10.2

pip install torch

1.12.0+cu102 torchvision

0.13.0+cu102 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch

1.12.0+cpu torchvision

0.13.0+cpu torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html

完成