如果电脑没有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来加载模型。找到这个修改位置主要根据报错的位置来找就可以找到类似的语句来进行修改了。