You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

torchvision0.8.1导入mobilenet_v3_small出现AttributeError如何解决

问题根因

你当前安装的torchvision 0.8.1版本不包含mobilenet_v3_small接口,该系列模型是在torchvision 0.9.0及更高版本才正式纳入torchvision.models模块的,所以运行时触发属性不存在报错。

解决方案

你可以根据自身环境需求选择以下任意一种方案解决:

  • 方案1:升级torch和torchvision到适配版本
    需注意torch和torchvision的版本严格对应,你当前使用CUDA11.0环境,可以安装与之适配的torch 1.8.x + torchvision 0.9.x版本组合,升级完成后原有导入代码可直接正常运行。
  • 方案2:不升级现有环境,本地复用官方实现
    如果你不想改动现有环境的依赖版本,可以直接将PyTorch官方开源的mobilenet_v3_small实现代码复制到你本地项目文件中,直接本地导入使用即可,无需依赖高版本torchvision。

内容的提问来源于stack exchange,提问作者user13074756

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.10.13 09:24:01