1 查看电脑安装的cuda版本

nvidia-smi

这里的红圈标注的是cuda 最高版本，虚拟环境中安装，只要不超过12.0就可以

第二步　去官网看torch, torchvision等版本的匹配关系