torch和torchvision对应版本匹配官网下载

https://download.pytorch.org/whl/torch_stable.html

ctrl+f找cp37,（因为下的是python3.7.10），（我的系统cuda版本是12.1,ubuntu22.04）

具体适配torch和torchvision参考

https://blog.csdn.net/qq_43481884/article/details/130684888?spm=1001.2014.3001.5506

环境要求

Python 3.7、CUDA ≥9.2、x86_64 架构、PyTorch ≥1.5.1、torchvision ≥0.6.1

看torch和torchvision部分

torch-1.13.0+cu116-cp37-cp37m-linux_x86_64.whl

torchvision-0.14.0+cu117-cp37-cp37m-linux_x86_64.whl

后来又出现一些问题，索性装python3.10

PyTorch 2.1.0+cu121（适配 Python 3.10 + CUDA 12.1）