This document is relevant for: Trn1, Trn2

Dockerfile for Application Container#

 1# Example pytorch neuron container
 2# To build:
 3#    docker build . -f Dockerfile.pt -t neuron-container:pytorch
 4# To run on EC2 Inf1 instances with AWS DLAMI:
 5#    docker run -it --net=host --device=/dev/neuron0 neuron-container:pytorch
 6
 7# You can find the latest Pytorch Training Image here - https://gallery.ecr.aws/neuron/pytorch-training-neuronx
 8FROM public.ecr.aws/neuron/pytorch-training-neuronx:2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04
 9RUN mkdir -p /opt/ml
10COPY model.py /opt/ml/model.py
11COPY mlp_train.py /opt/ml/mlp_train.py

This document is relevant for: Trn1, Trn2