Загрузить файл рассола, полученный из графического процессора в процессор ⇐ Python

[Anonymous]

Я столкнулся с проблемой при загрузке файла рассола в процессор. Я ищу это в Интернете, и они говорят, что мне нужно добавить параметр map_location. Однако после добавления этого параметра проблема все еще существует.

код выглядит следующим образом:

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

torch.__version__
torch.load('featurs.pkl',map_location='cpu')

>>>

'1.0.1.post2'
Attempting to deserialize object on a CUDA device
but torch.cuda.is_available() is False. If you are running
on a CPU-only machine, please use torch.load with map_location='cpu'
to map your storages to the CPU.

Я знаю, что это связано с разными устройствами, но я использую инструкцию в сообщении об ошибке, поэтому не знаю, как решить эту проблему на следующем шаге.< /p>

Заранее спасибо!


Подробнее здесь: https://stackoverflow.com/questions/570 ... gpu-to-cpu