本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。
使用 SageMaker AI 的自訂 Docker 容器
您可以調整現有的 Docker 映像以使用 SageMaker AI。當您的容器滿足預先建置的 SageMaker AI 映像目前不支援的功能或安全需求時,您可能需要使用現有的外部 Docker 映像與 SageMaker AI。有兩個工具組可讓您攜帶自己的容器,並加以調整以使用 SageMaker AI:
-
SageMaker Training Toolkit
– 使用此工具組搭配 SageMaker AI 進行訓練模型。 -
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 中。
架構 | 工具組原始程式碼 |
---|---|
TensorFlow |
|
MXNet |
|
PyTorch |
|
Chainer |