Abstract:Efficient GPU scheduling is the key to minimizing the execution time of the Deep Learning (DL) training workloads. DL training system schedulers typically allocate a fixed number of GPUs to each job, which inhibits high resource utilization and often extends the overall training time. The recent introduction of schedulers that can dynamically reallocate GPUs has achieved better cluster efficiency. This dynamic nature, however, introduces additional overhead by terminating and restarting jobs or requires modification to the DL training frameworks.We propose and develop an efficient, non-intrusive GPU scheduling framework that employs a combination of an adaptive GPU scheduler and an elastic GPU allocation mechanism to reduce the completion time of DL training workloads and improve resource utilization. Specifically, the adaptive GPU scheduler includes a scheduling algorithm that uses training job progress information to determine the most efficient allocation and reallocation of GPUs for incoming and running jobs at any given time. The elastic GPU allocation mechanism works in concert with the scheduler. It offers a lightweight and nonintrusive method to reallocate GPUs based on a “SideCar” process that temporarily stops and restarts the job’s DL training process with a different number of GPUs. We implemented the scheduling framework as plugins in Kubernetes and conducted evaluations on two 16-GPU clusters with multiple training jobs based on TensorFlow. Results show that our proposed scheduling framework reduces the overall execution time and the average job completion time by up to 45% and 63%, respectively, compared to the Kubernetes default scheduler. Compared to a termination based scheduler, our framework reduces the overall execution time and the average job completion time by up to 20% and 37%, respectively.

E-LAS: Design and Analysis of Completion-Time Agnostic Scheduling for Distributed Deep Learning Cluster

Hadar: Heterogeneity-Aware Optimization-Based Online Scheduling for Deep Learning Cluster

GPU Cluster Scheduling for Network-Sensitive Deep Learning

An Efficient and Non-Intrusive GPU Scheduling Framework for Deep Learning Training Systems.

Energy-Efficient GPU Clusters Scheduling for Deep Learning

ESG: Pipeline-Conscious Efficient Scheduling of DNN Workflows on Serverless Platforms with Shareable GPUs

ElasticBatch: A Learning-Augmented Elastic Scheduling System for Batch Inference on MIG

EaCO: Resource Sharing Dynamics and Its Impact on Energy Efficiency for DNN Training

Online Job Scheduling in Distributed Machine Learning Clusters

Scheduling Distributed Deep Learning Jobs in Heterogeneous Cluster with Placement Awareness

Deep Learning Workload Scheduling in GPU Datacenters: A Survey

Astraea: A Fair Deep Learning Scheduler for Multi-Tenant GPU Clusters

A Dual-Agent Scheduler for Distributed Deep Learning Jobs on Public Cloud Via Reinforcement Learning

Elastic Scheduler: Heterogeneous and Dynamic Deep Learning in the Cloud.

GAI: A Centralized Tree-Based Scheduler for Machine Learning Workload in Large Shared Clusters.

SCHED²: Scheduling Deep Learning Training Via Deep Reinforcement Learning.

Deep Learning Workload Scheduling in GPU Datacenters: Taxonomy, Challenges and Vision

Poster Abstract: Deep Learning Workloads Scheduling with Reinforcement Learning on GPU Clusters

TIAS: Two-level Information-Agnostic Job Scheduling in GPU Clusters

Learning to Optimize DAG Scheduling in Heterogeneous Environment

Communication Contention Aware Scheduling of Multiple Deep Learning Training Jobs