pytorch cpu与gpu load时相互转化
因为在gpu1上训练的模型,在另一个只有1个gpu的电脑上测试,遇到下面错误,才查找解决办法的。
raise RuntimeError('Attempting to deserialize object on CUDA device '
RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1. Please use torch.load with map_location to map your storages to an existing device.
pytorch将cpu训练好的模型参数load到gpu上,或者gpu->cpu