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.

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

SMDP-Based Dynamic Batching for Efficient Inference on GPU-Based Platforms

LBB: load-balanced batching for efficient distributed learning on heterogeneous GPU cluster

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

ElasticRoom: Multi-Tenant DNN Inference Engine Via Co-design with Resource-constrained Compilation and Strong Priority Scheduling

DeepBoot: Dynamic Scheduling System for Training and Inference Deep Learning Tasks in GPU Cluster

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

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

Elastic Deep Learning in Multi-Tenant GPU Clusters

Liquid: Intelligent Resource Estimation and Network-Efficient Scheduling for Deep Learning Jobs on Distributed GPU Clusters

Online Evolutionary Batch Size Orchestration for Scheduling Deep Learning Workloads in GPU Clusters

Dynamic Mini-batch SGD for Elastic Distributed Training: Learning in the Limbo of Resources

SchedTune: A Heterogeneity-Aware GPU Scheduler for Deep Learning

Energy-Efficient GPU Clusters Scheduling for Deep Learning

GPU Cluster Scheduling for Network-Sensitive Deep Learning

Adaptive Partitioning and Efficient Scheduling for Distributed DNN Training in Heterogeneous IoT Environment

Dynamic Space-Time Scheduling for GPU Inference

Deep Learning Research and Development Platform: Characterizing and Scheduling with QoS Guarantees on GPU Clusters.

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

BatOpt: Optimizing GPU-Based Deep Learning Inference Using Dynamic Batch Processing

DVABatch: Diversity-aware Multi-Entry Multi-Exit Batching for Efficient Processing of DNN Services on GPUs