Book Image

Accelerate Deep Learning Workloads with Amazon SageMaker

By : Vadim Dabravolski
Book Image

Accelerate Deep Learning Workloads with Amazon SageMaker

By: Vadim Dabravolski

Overview of this book

Over the past 10 years, deep learning has grown from being an academic research field to seeing wide-scale adoption across multiple industries. Deep learning models demonstrate excellent results on a wide range of practical tasks, underpinning emerging fields such as virtual assistants, autonomous driving, and robotics. In this book, you will learn about the practical aspects of designing, building, and optimizing deep learning workloads on Amazon SageMaker. The book also provides end-to-end implementation examples for popular deep-learning tasks, such as computer vision and natural language processing. You will begin by exploring key Amazon SageMaker capabilities in the context of deep learning. Then, you will explore in detail the theoretical and practical aspects of training and hosting your deep learning models on Amazon SageMaker. You will learn how to train and serve deep learning models using popular open-source frameworks and understand the hardware and software options available for you on Amazon SageMaker. The book also covers various optimizations technique to improve the performance and cost characteristics of your deep learning workloads. By the end of this book, you will be fluent in the software and hardware aspects of running deep learning workloads using Amazon SageMaker.
Table of Contents (16 chapters)
1
Part 1: Introduction to Deep Learning on Amazon SageMaker
6
Part 2: Building and Training Deep Learning Models
10
Part 3: Serving Deep Learning Models

Considering Hardware for Deep Learning Training

Training a large deep learning (DL) model is typically a lengthy and data- and resource-hungry process. Considering an extreme case of the GPT-3 NLP model, it took approximately 34 days to train it from scratch using 1,024 NVIDIA A100 GPUs. While it’s unlikely that you will have to train such a large model from scratch, even fine-tuning large DL models on your custom data can take days or even weeks.

Choosing a compute instance type for your specific model is a crucial step that will impact the cost and duration of training. AWS provides a wide spectrum of compute instances for various workload profiles. In this chapter, we will consider the price-performance characteristics of the most suitable instances for DL models, as well as scenarios where you should use one over the other for optimal performance.

Training large models also requires scaling training jobs across multiple GPU devices and compute instances, a process...