1.直接用如下版本的torch1.9.1+torchvision0.10.1会报如下错误:
2.官网PyTorch未找到对应cuda11.2版本的,因此直接安装cuda11.3版本
pip download torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
3.安装完成后查看版本,训练不出错