如果電腦沒有GPU,如何利用cpu來運行呢,沒有gpu的時候運行會報這個錯誤
RuntimeError: 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.
如何解決這個問題呢
解決:
將 model = torch.load(model_path)
改爲:
model = torch.load(model_path, map_location='cpu')
就是該torch.load裏的參數,將其指定爲cpu來加載模型。找到這個修改位置主要根據報錯的位置來找就可以找到類似的語句來進行修改了。