Ошибка атрибута: модуль «факел» не имеет атрибута «inference_mode»

#pytorch

Вопрос:

Я очень новичок в pytorch, и когда я пытаюсь запустить свой CNN, я столкнулся с ошибкой: AttributeError: у модуля «факел» нет атрибута «inference_mode». Кто-нибудь знает, что происходит? Это работало в Google colab, но больше нигде.

Комментарии:

1. Пожалуйста, укажите сегмент кода, в котором произошел сбой.

Ответ №1:

Скорее всего, это проблема с версией.

torch.inference_mode() недавно был добавлен в v1.9 . Убедитесь, что у вас есть правильная версия.

Попробуйте распечатать torch.__version__ , чтобы проверить свою версию.

Комментарии:

1. У меня все еще была проблема даже с torch 1.9.0. Я думаю, что исправил ее, обновив torchvision