搭配 Amazon SageMaker AI 使用 PyTorch 的資源

您可以使用 Amazon SageMaker AI，使用自訂 PyTorch 程式碼來訓練和部署模型。SageMaker AI Python SDK PyTorch 估算器和模型，以及 SageMaker AI 開放原始碼 PyTorch 容器，可讓您更輕鬆地撰寫 PyTorch 指令碼，並在 SageMaker AI 中執行指令碼。下一節提供參考資料，您可以用來了解如何將 PyTorch 與 SageMaker AI 搭配使用。

您想要做什麼？

我想要在 SageMaker AI 中訓練自訂 PyTorch 模型。

如需範例 Jupyter 筆記本，請參閱 Amazon SageMaker AI 範例 GitHub 儲存庫中的 PyTorch 範例筆記本。 GitHub

如需文件 , 請參閱使用 PyTorch 訓練模型。

我有我在 SageMaker AI 中訓練的 PyTorch 模型，而且我想要將它部署到託管端點。

如需詳細資訊，請參閱部署 PyTorch 模型。

我有在 SageMaker AI 之外訓練的 PyTorch 模型，而且我想要將它部署到 SageMaker AI 端點

如需詳細資訊，請參閱部署您自己的 PyTorch 模型。

我想要參閱 Amazon SageMaker Python SDK PyTorch 類別的 API 文件。

如需詳細資訊，請參閱 PyTorch 類別。

我想要尋找 SageMaker AI PyTorch 容器儲存庫。

如需詳細資訊，請參閱 SageMaker AI PyTorch Container GitHub 儲存庫。

我想要尋找 AWS 深度學習容器支援的 PyTorch 版本相關資訊。

有關詳細資訊，請參閱可用的深度學習容器映像。

如需編寫 PyTorch 訓練指令碼和搭配 SageMaker AI 使用 PyTorch 估算器和模型的一般資訊，請參閱搭配 SageMaker Python SDK 使用 PyTorch。