Adapting Your Own Docker Container to Work with SageMaker - Amazon SageMaker

Adapting Your Own Docker Container to Work with SageMaker

You can adapt an existing Docker image to work with SageMaker. You may need to use an existing, external Docker image with SageMaker when you have a container that satisfies feature or safety requirements that are not currently supported by a prebuilt SageMaker image. There are two toolkits that allow you to bring your own container and adapt it to work with SageMaker:

The following topics show how to adapt your existing image using the SageMaker Training and Inference toolkits:

Individual Framework Libraries

In addition to the SageMaker Training Toolkit and SageMaker Inference Toolkit, SageMaker also provides toolkits specialized for TensorFlow, MXNet, PyTorch, and Chainer. The following table provides links to the GitHub repositories that contain the source code for each framework and their respective serving toolkits. The instructions linked are for using the Python SDK to run training algorithms and host models on SageMaker. The functionality for these individual libraries is included in the SageMaker Training Toolkit and SageMaker Inference Toolkit.