python - 导入错误 : cannot import name 'mobilenet_v2' from 'torchvision.models'

标签 python pytorch fast-ai

我想在我的电脑上运行一个 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/

相关文章:

python - 从 PID 设置事件/前景窗口

python - 如何从unet_learner(fastai)获得正确的输出预测?

python-3.x - 在 fastai 库中使用 download_data() 和 untar_data()

deep-learning - PyTorch GRU 中隐藏状态的层顺序返回

machine-learning - 如何在fastai学习器上进行grid_search?

python - 在两个给定点之间画线(OpenCV,Python)

python - 鸢尾花数据集的多类分类

python - 在 Python/Numba 中访问数组会产生奇怪的结果

python - 如何使用 autograd 查找最小/最大点

python - 通过对最后 4 层求和来嵌入 BERT 句子