参考了网上各种说法,最终采用了torchvision和torch库版本不兼容的说法,完美运行!!
解决办法如下:
1、卸载原torchvision
pip uninstall torchvision
2、重新安装低版本的torchvision
pip install torchvision==0.4.1
这里注意以下,pip安装默认从国外镜像源下载,采用以上方式下载的话会非常的慢,不出意外的话会出现超时报错的现象。可以将下载源修改为清华的,代码如下:
pip install -i /simple torchvision==0.4.1
直接执行第二条代码就可以了,下载速度杠杠的!