从torchvision.models.vgg导入model_urls失败的解决方法及适配版本咨询

版本变更说明 #

• 在torchvision 0.11.x及更早版本中，model_urls直接暴露在torchvision.models.vgg模块下，可直接导入使用。
• 从torchvision 0.12.x开始，官方重构了预训练权重管理逻辑，model_urls被移至内部私有模块，不再对外提供直接导入的接口，这也是降级到0.12、0.13版本仍报错的原因。

可行解决方法 #

方法一：适配新版本的标准权重获取方式

无需依赖model_urls，使用torchvision 0.12+推荐的API获取预训练权重及URL，示例代码：

from torchvision.models import vgg16
from torchvision.models.vgg import VGG16_Weights

# 获取预训练权重的URL
weights = VGG16_Weights.DEFAULT
weight_url = weights.url

# 加载带预训练权重的模型
model = vgg16(weights=VGG16_Weights.DEFAULT)

这种方式兼容性更强，也符合官方后续的迭代方向。

方法二：降级到兼容旧接口的版本

如果必须直接使用model_urls，需降级至torchvision 0.11.3及更早版本，同时要保证PyTorch与torchvision版本匹配（如PyTorch 1.10.x对应torchvision 0.11.x），安装命令示例：

pip install torchvision==0.11.3 torch==1.10.2

内容的提问来源于stack exchange，提问作者Pooyan Rezaeipour