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

Pytorch+cuda+torchvision的安装命令

python安装对应版本的cuda和torch以及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下,复制命令即可!

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

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

完成

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年3月29日
下一篇 2023年3月29日

相关推荐