问题描述
我运行一个python代码,就报这个错误
ModuleNotFoundError: No module named 'torchvision.models.utils'
问题原因
根据项目说明文档,需要 torch==1.2.0。但是我的环境比较新,torch==1.13.1+cu116,torchvision==0.14.1+cu116。所以是因为torchvision版本过高导致的错误。
解决方法
将from torchvision.models.utils import load_state_dict_from_url
改为:from torch.hub import load_state_dict_from_url