我想在我的电脑上运行一个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。如果有人需要更多的规范,我会重新编辑这段代码。谢谢。
1条答案
按热度按时间ac1kyiln1#
我刚刚完成修复这个问题与我的系统。卸载任何pytorch,torchvision由conda和pip。卸载fastai以及。
转到https://pytorch.org/get-started/locally/并根据您的cuda版本等运行conda命令。然后