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
相关产品推荐
相关产品推荐

