model = torch.load(model_path, map_location='cpu')

没有cuda时，记得加上map_location='cpu'，要不然会报错