ImportError: невозможно импортировать имя mobilenet_v2 из torchvision.models

Question

Я хочу запустить модель глубокого обучения fastai на моем p c. Не тренировать, а просто запустить предварительно обученную модель на моем P C. У меня есть файл .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, а именно c, IDE spyder, подключенного к моей среде anaconda. Я отредактирую это, если кому-то понадобятся дополнительные спецификации. Спасибо.

Answer 1

Я только что закончил исправлять эту проблему в своей системе. Удалите все pytorch, torchvision by conda и pip. Удалите также fastai.

Go до https://pytorch.org/get-started/locally/ и запустите команду conda на базе вашей версии cuda и et c. Тогда

conda install -c fastai fastai