1. 问题描述
2.问题分析
-
torch和torchvision版本不同,比如torch是支持cuda的,但是torchvision不支持cuda
-
torch和cuda版本不对
我的torch版本是1.31+cuda117,而torchvision是cpu版本的,因此报错,需要重新安装torchvision
3. 解决方法
3.1 卸载torchvision
pip uninstall torchvision
3.2 下载torchvision
https://download.pytorch.org/whl/torch_stable.html
3.3 安装torchvision
pip install torchvision-0.14.0+cu117-cp37-cp37m-win_amd64.whl