Ad

SRExpert

Stop juggling 6 Kubernetes tools.

Monitoring, alerts, security, Helm deploys and AI ops in one platform.

Start a 30-day free trial

SageMaker Inference now supports Multi-Instance GPU to maximize GPU utilization for model endpoints

6d

Age

Published
Collected

Amazon SageMaker Inference now supports GPU partitioning with NVIDIA Multi-Instance GPU (MIG), which divides one GPU into hardware-isolated partitions so you can run a different model in each or take a right-sized fraction of a newer accelerator. This post shows how two optional API fields place one model per partition, and what we measured against a whole older GPU.

Excerpt from the source

When you deploy a model to a GPU endpoint, the smallest unit of GPU you can request is one whole accelerator. This constraint can cost you twice. First, you can’t host more models on an instance than it has accelerators. An ml.p4d.24xlarge has eight A100s, so it holds eight models, whether each one needs 40 GB or 14 GB. Second, and less obvious, it makes every deployment come down to a choice of which full GPU to deploy to. As your concurrency and prompt lengths grow, a full GPU on an older…

overfeed.news indexes and links. We publish a short excerpt — the full article stays at AWS Machine Learning Blog.

Log in to follow this source
SageMaker Inference now supports Multi-Instance GPU to maximize…