pytorch 导入错误:无法从"torchvision. models"导入名称"mobilenet_v2"

f4t66c6m  于 2023-03-02  发布在  其他
关注(0)|答案(1)|浏览(486)

我想在我的电脑上运行一个fastai深度学习模型。不是train,只是在我的电脑上运行预训练模型。我有.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。如果有人需要更多的规范,我会重新编辑这段代码。谢谢。

ac1kyiln

ac1kyiln1#

我刚刚完成修复这个问题与我的系统。卸载任何pytorch,torchvision由conda和pip。卸载fastai以及。
转到https://pytorch.org/get-started/locally/并根据您的cuda版本等运行conda命令。然后

conda install -c fastai fastai

相关问题