torch与torchvision版本对应

你可能有时会遇到一下错误

AssertionError: Torch not compiled with CUDA enabled
或者是你的gpu不可用的情况

有可能是你的torch与torchvision版本冲突
或者是你的torch安装的是cpu的版本
可以用pip list命令查询
如果是

torch 1.12.1+cpu
torchvision 0.13.1+cpu

显而易见，这是cpu的版本
那如何重装一下呢？
首先卸载

pip uninstall torch 
pip uninstall torchvision
pip uninstall torchaudio

查看网址：https://pytorch.org/get-started/locally/
特别注意，要选择与自己机器上装载的CUDA一样的版本
（https://blog.csdn.net/littleyy666/article/details/127464216这篇文章有讲【tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2】的安装，适合换电脑选手）
复制下面的命令，到自己本地运行即可

pip list查看安装成功