我想在我的电脑上运行一个 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 中运行此代码,具体来说,是连接到我的 anaconda 环境的 spyder IDE。如果有人需要更多规范,我会重新编辑它。谢谢你。
最佳答案
我刚刚用我的系统解决了这个问题。通过 conda 和 pip 卸载任何 pytorch、torchvision。也卸载fastai。
转至 https://pytorch.org/get-started/locally/并根据您的 cuda 版本等在那里运行 conda 命令。然后
conda install -c fastai fastai
关于python - 导入错误 : cannot import name 'mobilenet_v2' from 'torchvision.models' ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61666911/