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.

Related resources

• Getting started with Neuron DLC using Docker — Set up Docker on EC2 for Neuron

• Customize Neuron DLC — Customize a pre-built Deep Learning Container

• Neuron Deep Learning Containers — Find pre-built DLC images in ECR