多GPU实现同时训练模型

1.1单GPU：
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
说明：设置单块GPU编号
1.2多GPU：

device_ids = [0,2,4,6]
#说明：GPU编号，一块显卡集成两块GPU，则同一块卡的编号不能同时出现。
model = nn.DataParallel(model, device_ids=device_ids).cuda()
1.3实例：在alphapose训练中使用多线程训练，
1.修改文件：主程序train.py，加入上述命令：