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