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
完成
文章出处登录后可见!
已经登录?立即刷新