[发明专利]分配GPU卡的数量的方法、装置、电子设备和存储介质

权利要求书

1.一种分配GPU卡的数量的方法，其特征在于，包括：

获取图形处理器GPU卡的类型，根据所述图形处理器GPU卡的类型注册资源名字；

获取用户提交的任务启动参数，所述任务启动参数包括：用户使用的图形处理器GPU卡的类型以及用户需要的图形处理器GPU卡的数量；

采用资源最大化利用的原则，根据所述资源名字和所述任务启动参数为所述任务分配图形处理器卡的数量。

2.根据权利要求1所述的分配GPU卡的数量的方法，其特征在于，所述采用资源最大化利用的原则为所述任务分配图形处理器卡的数量，包括：

根据所述资源名字查询每个候选节点可用的总图形处理器GPU卡数，以及当前任务使用的图形处理器GPU数目，确定所述每个候选节点剩余的图形处理器GPU卡的数量，为所述任务分配图形处理器卡的数量。

3.根据权利要求2所述的分配GPU卡的数量的方法，其特征在于，所述根据所述资源名字查询每个候选节点可用的总图形处理器GPU卡的数量，以及当前任务使用的图形处理器GPU卡的数量，确定每个候选节点剩余的图形处理器GPU卡的数量，为所述任务分配图形处理器卡的数量，包括：

当确定用户需要的图形处理器GPU卡的数量大于集群剩余的图形处理器GPU卡的数量时，将所述任务加入等待队列中；

当确定用户需要的图形处理器GPU卡的数量小于所述集群剩余的图形处理器GPU卡的数量时，根据剩余最少的节点原则分配所述图形处理器GPU卡的数量。

4.根据权利要求3所述的分配GPU卡的数量的方法，其特征在于，所述方法还包括：

获取所述集群各个节点的图形处理器GPU卡的数量，中央处理器以及内存；

根据预设比值的加权比分别对所述集群各个节点做加权打分；

根据加权比高的组合分配所述图形处理器GPU卡的数量。

5.根据权利要求4所述的分配GPU卡的数量的方法，其特征在于，所述方法还包括：

根据所述用户需要的图形处理器GPU卡的类型，向所述节点申请GPU卡的类型对应的资源；

调度所述任务到所述节点，发送消息至所述节点的代理，启动所述任务；

所述节点的代理运行所述任务并发送所述任务的状态至所述集群。

6.一种分配图形处理器GPU卡的数量的装置，其特征在于，包括：

注册资源名字模块：用于获取图形处理器GPU卡的类型，根据所述图形处理器GPU卡的类型注册资源名字；

获取模块：用于获取用户提交的任务启动参数，所述任务启动参数包括：用户使用的图形处理器GPU卡的类型以及用户需要的图形处理器GPU卡的数量；

分配模块：用于采用资源最大化利用的原则，根据所述资源名字和所述任务启动参数为所述任务分配图形处理器卡的数量。

7.根据权利要求1所述的分配GPU卡的数量的装置，其特征在于，所述分配模块，还用于：

根据所述资源名字查询所述每个候选节点可用的总图形处理器GPU卡数，以及当前任务使用的图形处理器GPU数目，确定所述每个候选节点剩余的图形处理器GPU卡的数量，为所述任务分配所述图形处理器卡的数量。

8.根据权利要求7所述的分配GPU卡的数量的装置，其特征在于，所述分配模块还用于：

当确定用户需要的图形处理器GPU卡的数量大于集群剩余的图形处理器GPU卡的数量时，将所述任务加入等待队列中；

当确定用户需要的图形处理器GPU卡的数量小于所述集群剩余的图形处理器GPU卡的数量时，根据剩余最少的节点原则分配图形处理器GPU卡的数量。

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

所述处理器通过调用所述存储器存储的程序或指令，用于执行如权利要求1至5任一项所述分配GPU卡的数量的方法。

10.一种计算机可读存储介质，其特征在于，所述非暂态计算机可读存储介质存储程序或指令，所述程序或指令使计算机执行如权利要求1至5任一项所述分配GPU卡的数量的方法。