多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).cud

>>阅读原文<<