Latest SageMaker AI Operators for Kubernetes - Amazon SageMaker AI

Latest SageMaker AI Operators for Kubernetes

This section is based on the latest version of SageMaker AI Operators for Kubernetes using AWS Controllers for Kubernetes (ACK).

Important

If you are currently using version v1.2.2 or below of SageMaker Operators for Kubernetes, we recommend migrating your resources to the ACK service controller for Amazon SageMaker. The ACK service controller is a new generation of SageMaker Operators for Kubernetes based on AWS Controllers for Kubernetes (ACK).

For information on the migration steps, see Migrate resources to the latest Operators.

For answers to frequently asked questions on the end of support of the original version of SageMaker Operators for Kubernetes, see Announcing the End of Support of the Original Version of SageMaker AI Operators for Kubernetes

The latest version of SageMaker AI Operators for Kubernetes is based on AWS Controllers for Kubernetes (ACK), a framework for building Kubernetes custom controllers where each controller communicates with an AWS service API. These controllers allow Kubernetes users to provision AWS resources like databases or message queues using the Kubernetes API.

Use the following steps to install and use ACK to train, tune, and deploy machine learning models with Amazon SageMaker AI.

Install SageMaker AI Operators for Kubernetes

To set up the latest available version of SageMaker AI Operators for Kubernetes, see the Setup section in Machine Learning with the ACK SageMaker AI Controller.

Use SageMaker AI Operators for Kubernetes

For a tutorial on how to train a machine learning model with the ACK service controller for Amazon SageMaker AI using Amazon EKS, see Machine Learning with the ACK SageMaker AI Controller.

For an autoscaling example, see Scale SageMaker AI Workloads with Application Auto Scaling

Reference

See also the ACK service controller for Amazon SageMaker AI GitHub repository or read AWS Controllers for Kubernetes Documentation.