【安装】各平台下安装低版本CUDA的PyTorch

目前直接点开官网,看到的安装链接是这样的:
【安装】各平台下安装低版本CUDA的PyTorch_第1张图片
倘若你的系统的CUDA版本恰好是11.2,而且又懒得升级,可以采取指定版本安装的方式装PyTorch:

运行指令,查看自己的显卡的版本号:

nvidia-smi -i 0

打开官网的所有版本。

注意:PyTorch没有专门针对CUDA 11.2发行的版本,如果要下载,只能下载CUDA 11.1对应的。
而CUDA 11.1对应的版本,不包括torch 1.11。

如果想下torch 1.11.0+cu111,那是不可能的。

搜索cu111会发现cu111不支持pytorch 1.11,其他的倒无所谓。

然后,手动指定安装版本就行,如下:

pip --default-timeout=1000 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(科普,pytorch,深度学习,人工智能)