Пожалуйста, используйте torch.load с map_location=torch.device('cpu') ⇐ Python

[Anonymous]

Я запускаю программу Python, но у меня нет графического процессора. Что мне сделать, чтобы Python использовал процессор вместо графического процессора?

Код: Выделить всё

$ python extract_feature.py --data mnist --net checkpoint_4.pth.tar --features pretrained

Это дает мне следующее предупреждение:

=> RuntimeError: попытка десериализации объекта на устройстве CUDA, но факел .cuda.is_available() имеет значение false. Если вы работаете на машине только с процессором, используйте torch.load с map_location=torch.device('cpu') для сопоставления ваших хранилищ с процессором.

На фотографии представлена ​​структура моего проекта Python:



Подробнее здесь: https://stackoverflow.com/questions/658 ... -devicecpu