調整您自己的 Docker 容器以使用 SageMaker

您可以調整現有的 Docker 映像以使用 SageMaker。當您的容器符合預 SageMaker先建置映像目前不支援的功能或安全需求 SageMaker 時，您可能需要使用現有的外部 Docker 映像檔。有兩個工具包可讓您攜帶自己的容器並將其適應以使用 SageMaker：

• SageMaker 訓練工具組 — 使用此工具組來訓練模型 SageMaker。

• SageMaker 推論工具組 — 使用此工具組來部署. SageMaker

下列主題說明如何使用「 SageMaker 訓練與推論」工具組調整現有影像：

主題

• 個別架構程式庫
• 使用 SageMaker 訓練與推論工具組
• 使用自有訓練容器
• 為 Amazon 調整您自己的推論容器 SageMaker

個別架構程式庫

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

架構	工具組原始程式碼
TensorFlow	SageMaker TensorFlow 訓練 SageMaker TensorFlow 服務
MXNet	SageMaker MXNet 訓 SageMaker MXNet 論
PyTorch	SageMaker PyTorch 訓練 SageMaker PyTorch 推論
Chainer	SageMaker 連鎖容器 SageMaker