找不到名为'torchvision.models.utils'的模块。

Question

当我使用pytorch=1.10.0，torchvision=0.11.1的环境运行代码时，遇到了从torchvision.models.utils import load_state_dict_from_url这个语句开始的问题。会出现以下错误：

>>> from torchvision.models.utils import load_state_dict_from_url
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torchvision.models.utils'

- Oscar Rangel

Answer 1

在查阅 torchvision 的代码库后，找到了一个解决方案：

需要注意的是，这个语法只适用于较高版本的 PyTorch。

原来的代码 .utils import load_state_dict_from_url 不再适用。 你不能从 .utils 导入 load_state_dict_from_url。

把 .utils 改成 torch.hub 可以解决这个问题。

from torch.hub import load_state_dict_from_url

这对我有效。

- Oscar Rangel