ModuleNotFoundError: No module named ‘torchvision.models.utils‘

时间:2024-01-25 17:45:53

问题描述

我运行一个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