Docker examples#

This section provides example Dockerfiles and supporting code for building Neuron containers. Use these as starting points for your own container builds, or reference them when customizing a Deep Learning Container.

Inference Dockerfiles

Dockerfiles for building inference containers, including DLC-based, TorchServe, and TensorFlow Serving examples.

Training Dockerfiles

Dockerfiles for building training containers on Trainium, including a complete MLP training example.

Device plugin Dockerfile

The Dockerfile.device-plugin in this directory shows how to build the Neuron device plugin container for Kubernetes.