Amazon SageMaker
Developer Guide

Use TensorFlow with Amazon SageMaker

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

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

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