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.

Design and Performance Modeling of A YARN-based GPU Resource Scheduling System

Deep Learning Workload Scheduling in GPU Datacenters: A Survey

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

Performance Modeling and Evaluation of Distributed Deep Learning Frameworks on GPUs

Towards Universal Performance Modeling for Machine Learning Training on Multi-GPU Platforms

A Virtual Multi-Channel GPU Fair Scheduling Method for Virtual Machines.

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

Performance Modeling and Analysis of Distributed Deep Neural Network Training with Parameter Server

Energy-Efficient GPU Clusters Scheduling for Deep Learning

Characterization and Prediction of Deep Learning Workloads in Large-Scale GPU Datacenters

A Topology-Aware Performance Prediction Model for Distributed Deep Learning on GPU Clusters

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

An Empirical Study on Low GPU Utilization of Deep Learning Jobs

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

Horus: Interference-Aware and Prediction-Based Scheduling in Deep Learning Systems

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

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

Effective Scheduler for Distributed DNN Training Based on MapReduce and GPU Cluster

Scheduling Deep Learning Jobs in Multi-Tenant GPU Clusters via Wise Resource Sharing

GPU Shared Scheduling System Under Deep Learning Container Cloud Platform

HPH: Hybrid Parallelism on Heterogeneous Clusters for Accelerating Large-scale DNNs Training.