← 返回列表
GPU系统的控制方法、装置、电子设备和存储介质
摘要文本
本公开涉及一种GPU系统的控制方法、装置、电子设备和存储介质。所述GPU系统包括至少两个GPU，所述方法包括：获取用于训练目标模型的训练样本集；对于所述GPU系统中的任一GPU，获得所述GPU的目标功耗和当前功耗；响应于进入新的训练样本分配周期，根据所述目标功耗和所述当前功耗，确定所述GPU对应的目标批大小，并向所述GPU分配所述目标批大小对应的目标训练样本批次，以通过所述GPU根据所述目标训练样本批次训练所述目标模型。
申请人信息
- 申请人:摩尔线程智能科技(北京)有限责任公司
- 申请人地址:100080 北京市海淀区翠微中里14号楼四层B655
- 发明人: 摩尔线程智能科技(北京)有限责任公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | GPU系统的控制方法、装置、电子设备和存储介质 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311550223.0 |
| 申请日 | 2023/11/20 |
| 公告号 | CN117539631A |
| 公开日 | 2024/2/9 |
| IPC主分类号 | G06F9/50 |
| 权利人 | 摩尔线程智能科技(北京)有限责任公司 |
| 发明人 | 请求不公布姓名 |
| 地址 | 北京市海淀区翠微中里14号楼四层B655 |
专利主权项内容
1.一种GPU系统的控制方法，其特征在于，所述GPU系统包括至少两个GPU，所述方法包括：获取用于训练目标模型的训练样本集；对于所述GPU系统中的任一GPU，获得所述GPU的目标功耗和当前功耗；响应于进入新的训练样本分配周期，根据所述目标功耗和所述当前功耗，确定所述GPU对应的目标批大小，并向所述GPU分配所述目标批大小对应的目标训练样本批次，以通过所述GPU根据所述目标训练样本批次训练所述目标模型。