Prebuilt SageMaker Docker Images for Deep Learning
SageMaker provides prebuilt Docker images that include deep learning framework libraries and
other dependencies needed for training and inference. For a complete list of the available
pre-built Docker images, see Deep Learning Containers Images
If you are not using the Amazon SageMaker Python SDK
Using the SageMaker Python SDK
With the SageMaker Python SDK
Framework | Instructions |
---|---|
TensorFlow |
|
MXNet |
|
PyTorch |
|
Chainer |
|
Hugging Face |
Extending Prebuilt SageMaker Docker Images
You can customize these prebuilt containers or extend them to handle any additional
functional requirements for your algorithm or model that the prebuilt SageMaker Docker image
doesn't support. For an example, see Extending Our PyTorch Containers
You can also use prebuilt containers to deploy your custom models or models that have been
trained in a framework other than SageMaker. For an overview of the process of bringing the
trained model artifacts into SageMaker and hosting them at an endpoint, see Bring Your Own Pretrained MXNet or TensorFlow Models into Amazon
SageMaker