使用 SageMaker AI 的自訂 Docker 容器 - Amazon SageMaker AI

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

使用 SageMaker AI 的自訂 Docker 容器

您可以調整現有的 Docker 映像以使用 SageMaker AI。當您的容器滿足預先建置的 SageMaker AI 映像目前不支援的功能或安全需求時,您可能需要使用現有的外部 Docker 映像與 SageMaker AI。有兩個工具組可讓您攜帶自己的容器,並加以調整以使用 SageMaker AI:

以下主題說明如何使用 SageMaker 訓練和推論工具組來調整現有映像:

個別架構程式庫

除了 SageMaker Training Toolkit 和 SageMaker AI Inference Toolkit 之外,SageMaker AI 還提供專門用於 TensorFlow、MXNet、PyTorch 和 Chainer 的工具組。以下資料表提供 GitHub 儲存庫的連結,儲存庫內包含每個架構及其各自的服務工具組的原始程式碼。連結的指示適用於使用 Python SDK 在 SageMaker AI 上執行訓練演算法和託管模型。這些個別程式庫的功能包含在 SageMaker AI Training Toolkit 和 SageMaker AI Inference Toolkit 中。