一种YARN集群GPU资源调度方法、装置和介质

摘要

本发明公开了一种YARN集群GPU资源调度方法、装置和介质，用以在YARN集群中实现GPU资源调度的同时，降低实现GPU资源的调度的复杂度。所述YARN集群GPU资源调度方法，为YARN集群中包含的GPU节点添加GPU标签；所述方法，包括：接收需要调度GPU资源的任务，所述任务中携带有完成所述任务所需的GPU资源数量；根据所述GPU标签确定所述YARN集群中可用的GPU节点并查询每一可用GPU节点的剩余GPU资源数量；根据完成所述所需的GPU资源数量和每一可用GPU节点的剩余GPU资源数量，确定GPU资源调度策略；根据确定出的资源调度策略为所述任务调度GPU资源。