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

Question

我想在我的电脑上运行fastai深度学习模型。不进行训练，只需在PC上运行预先训练的模型即可。我有.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

我刚刚用系统解决了这个问题。通过conda和pip卸载任何pytorch，torchvision。也要卸载fastai。

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

conda install -c fastai fastai