ImportError：无法从“torchvision.models”导入名称“mobilenet_v2”

Question

我想在我的电脑上运行 fastai 深度学习模型。不训练，只需在我的电脑上运行预训练的模型。我有 .pth 文件。我尝试导入我安装的 fastai 模块，但收到错误：

ImportError: cannot import name 'mobilenet_v2' from 'torchvision.models' (C:\file_path\__init__.py)

我尝试执行的代码：

#From the fastai library
from fastai import *
from torchvision.models import * 
from fastai.vision import *

我找不到任何解决方案来解释为什么会出现此错误。

我在 anaconda 中运行此代码，具体来说，spyder IDE 连接到我的 anaconda 环境。如果有人需要更多规格，我将重新编辑此内容。谢谢。

Answer 1

我刚刚用我的系统修复了这个问题。卸载任何 pytorch、torchvision by conda 和 pip。也卸载 fastai。

转到https://pytorch.org/get-started/locally/并根据您的 cuda 版本等运行 conda 命令。然后

conda install -c fastai fastai