[发明专利]一种程序包部署方法、电子设备及分布式系统

权利要求书

1.一种程序包部署方法，其特征在于，应用于分布式系统中的第一计算节点，所述方法包括：

接收训练任务信息，所述训练任务信息中包含执行训练任务的各台计算节点信息；

根据所述训练任务信息，确定所述第一计算节点的状态是否为主用状态；

如果为主用状态，获取训练程序包，并将所获取的训练程序包部署到所述执行训练任务的各台计算节点。

2.根据权利要求1所述的方法，其特征在于，在接收训练任务信息之后，所述方法还包括：

解析所述训练任务信息，得到训练程序包的存储地址、执行训练任务的各台计算节点的状态信息及设备地址；

所述根据所述训练任务信息，确定所述第一计算节点的状态是否为主用状态，包括：

在所述各台计算节点的状态信息中，查找所述第一计算节点的设备地址对应的状态信息；

确定所查找到的状态信息是否为主用状态；

所述获取训练程序包，包括：

根据所述训练程序包的存储地址，获取所述训练程序包；

所述将所获取的训练程序包部署到所述执行训练任务的各台计算节点，包括：

根据所述各台计算节点的设备地址，在所述各台计算节点中部署所述训练程序包。

3.根据权利要求1所述的方法，其特征在于，所述方法还包括：

如果所述第一计算节点的状态为主用状态，检测到所述执行训练任务的各台计算节点均成功部署所述训练程序包后，生成标记文件，并将所述标记文件发送至所述各台计算节点。

4.根据权利要求1所述的方法，其特征在于，所述方法还包括：

如果所述第一计算节点的状态为主用状态，检测到存在部署所述训练程序包失败的计算节点后，输出用于提示部署失败的第一提示信息。

5.根据权利要求3所述的方法，其特征在于，所述方法还包括：

如果所述第一计算节点的状态不为主用状态，判断是否在预设时间段内接收到所述标记文件；

如果否，输出用于提示部署失败的第二提示信息。

6.根据权利要求1所述的方法，其特征在于，所述方法还包括：

在所述执行训练任务的各台计算节点均成功部署所述训练程序包后，运行所述训练程序包，进行数据训练。

7.根据权利要求1-6任一项所述的方法，其特征在于，所述将所获取的训练程序包部署到所述执行训练任务的各台计算节点，包括：

通过无限带宽技术Infiniband，将所获取的训练程序包部署到所述执行训练任务的各台计算节点。

8.一种电子设备，其特征在于，包括：存储器和处理器，其中，

存储器，用于存放计算机程序；

处理器，用于执行存储器上所存放的程序时，实现权利要求1-7任一所述的方法步骤。

9.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质内存储有计算机程序，所述计算机程序被处理器执行时实现权利要求1-7任一所述的方法步骤。

10.一种分布式系统，其特征在于，包括：至少两台计算节点；

所述计算节点，用于接收训练任务信息，所述训练任务信息中包含执行训练任务的各台计算节点信息；根据所述训练任务信息，确定自身状态是否为主用状态；如果为主用状态，获取训练程序包，并将所获取的训练程序包部署到所述执行训练任务的各台计算节点。