主要原因是torchvision的版本与pytorch版本不兼容,选择和pytorch版本兼容的torchvision版本即可。
一、查看pytorch的版本:在python IDE中输入,
import torch
print(torch.__version__)
可以看到我的pytorch版本为1.7.1:

二、选择相匹配的torchvision版本:
可以到pytorch官网上根据自己安装的pytorch确认应该安装的torchvision版本,比如pytorch1.7.1就应该对应torchvision0.8.2:

这样就可以成功安装啦!
版权声明:本文为aibeeeee原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。