基于GPU共享的深度学习训练任务加速调度框架OA

Deep learning(DL)is increasingly being applied across a wide range of business scenarios.How to efficiently utilize resources in GPU clusters for training DL tasks and reduce task completion times has garnered sustained attention from both industry and academia.A single DL training task often fails to fully leverage all the computational resources of a GPU,and the exclusive GPU allocation by traditional schedulers leads to low resource utilization.This paper proposes a GPU-sharing-based task scheduling framework,G-Share,which allows multiple DL tasks to be trained on the same GPU simul-taneously,enabling co-location scheduling.Task scheduling and resource allocation are performed while being aware of the interference between co-located tasks,aiming to enhance GPU utilization and thereby accelerate task execution.Specifically,it first characterizes the mutual interference information between tasks through offline modeling and online updates,and models the GPU-sharing-based scheduling prob-lem as a weighted bipartite graph minimum matching problem.By solving this problem,resource allocation results are obtained,and a dynamic task scheduling mechanism combined with time-slicing is employed to perceive changes in the optimal co-location combinations of tasks in online scenarios.Ex-periments conducted on the DL task workload data from SenseTime demonstrates that G-Share achieves a 20.6%reduction in the average task completion times compared to benchmark methods.