[python][pytorch]多GPU下的模型保存与加载

说明 在模型训练的时候，往往使用的是多GPU的环境；但是在模型验证或者推理阶段，往往使用单GPU甚至CPU进行运算。那么中间有个保存和加载的过程。下面来总结一下。 多GPU进行训练 首先设置可见的GPU数量，有两种方式可以声明： 在shell脚本中声明： export CUDA_VISIBLE_DE
posted @ 2020-06-01 15:13  wildkid1024  阅读(4630)  评论(0)  编辑  收藏  举报