Amazon SageMaker
Developer Guide

Using PyTorch with Amazon SageMaker

You can use Amazon SageMaker to train and deploy a model using custom PyTorch code. Amazon SageMaker provides an open-source container that makes writing a PyTorch script and running it in Amazon SageMaker easier. For information about how to build and use the Amazon SageMaker PyTorch container, see the GitHub repository at https://github.com/aws/sagemaker-pytorch-container.

For information about PyTorch versions supported by the Amazon SageMaker PyTorch container, see https://github.com/aws/sagemaker-python-sdk#pytorch-sagemaker-estimators.

For information about writing PyTorch training scripts and using PyTorch estimators with Amazon SageMaker, see https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/pytorch/README.rst.