错误笔记1
报错:
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=torch.device(‘cpu’) to map your storages to the CPU.
可能的原因:gpu训练的模型保存后,在无gpu设备上无法直接加载
解决方法:
需要设置map_location="cpu"
路径在:/home/xxx/.local/lib/python3.6/site-packages/torch/serialization.py
486行def load(f, map_location=“cpu”, pickle_module=pickle, **pickle_load_args):