Amazon SageMaker
Developer Guide

Use PyTorch with Amazon SageMaker

You can use Amazon SageMaker to train and deploy a model using custom PyTorch code. The Amazon SageMaker Python SDK PyTorch estimators and models and the Amazon SageMaker open-source PyTorch container make writing a PyTorch script and running it in Amazon SageMaker easier. For information about writing PyTorch training scripts and using PyTorch estimators and models with Amazon SageMaker, see https://github.com/aws/sagemaker-python-sdk#pytorch-sagemaker-estimators.

For information about PyTorch versions supported by the Amazon SageMaker PyTorch container, see https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/pytorch/README.rst.

For information about how to build and contribute to the Amazon SageMaker PyTorch container, see the GitHub repository at https://github.com/aws/sagemaker-pytorch-container.