多GPU实现同时训练模型

2022-10-23 12:07:14

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，加入上述命令：
在这里插入图片描述

  • 作者：角落落脚地
  • 原文链接：https://blog.csdn.net/weixin_42747686/article/details/88694229
    更新时间：2022-10-23 12:07:14