因为是老版本,需要定位
直接根据官网定位
在对应Linux虚拟环境中直接输入命令
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
检查是否安装成功
在pytorch环境下输入逐次输入以下命令
python #检查python版本
import torch
torch.cuda.is_available()
显示结果见下:(则表明已正确安装pytorch)
True