TensorFlow

TensorFlow DLCs come in two flavors, each with its own guide:

• Training — train models with the TensorFlow framework on Amazon SageMaker AI, with EFA-capable multi-node support on Amazon Linux 2023.
• Inference — serve TensorFlow SavedModel artifacts with TensorFlow Serving on Amazon SageMaker AI, in CPU and GPU variants.

The two are built from separate images and separate ECR repositories (tensorflow-training and tensorflow-inference), so pick the guide that matches your workload.