RuntimeError: ожидаемый байт процессора и dtype устройства, но получил процессор и бул dtype

Question

Как описано в проблема, которую я открыл , я получаю следующую ошибку при запуске модели Pytorch обратного приготовления на ЦП:

RuntimeError: expected device cpu and dtype Byte but got device cpu and dtype Bool

Я попытался запустить файл demo.ipynb в потоках Intel i7-4700HQ 8 моего ноутбука и на моем настольном компьютере Ryzen 3700x. Я использовал Arch Linux на своем ноутбуке и Manjaro на рабочем столе.

Модель работает нормально, когда я запускаю ее на графическом процессоре Google Collabs.

Согласно файлу demo.ipynb модель должна бытьможет работать на процессоре. Кто-нибудь знает, нужно ли настраивать какие-либо параметры, чтобы он работал?

Answer 1

Как заявили @iacolippo и в сеансе комментариев и myDennisCode , проблема действительно заключалась в зависимостях версий. У меня было torchvision==0.4.0 (что меня смутило) и torch==1.2.0.

Чтобы решить проблему, просто установите torch==0.4.1 и torchvision==0.2.1.