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

你可能感兴趣的:(工具安装,经验总结,python,深度学习,pytorch)