@@ -14,23 +14,23 @@ AWS Neuron Deep Learning Containers (DLCs) are a set of Docker images for traini
1414
1515| Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | Other Packages |
1616| -----------------------------------------------------------------------------------------------------------------------------------------| -----------------------------------------------------------------------------| --------------------| ------------------------------| ------------------------| --------------------------------------------------------------------------------------------| -------------------|
17- | [ PyTorch 2.5.1 ] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.22 .0/docker/pytorch/inference/2.5.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_inference, torch-neuronx, transformers-neuronx | Neuron 2.22 .0 | trn1,trn2,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.5.1 -neuronx-py310-sdk2.22 .0-ubuntu22.04 | torchserve 0.11.0 |
17+ | [ PyTorch 2.6.0 ] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.23 .0/docker/pytorch/inference/2.5.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_inference, torch-neuronx, transformers-neuronx | Neuron 2.23 .0 | trn1,trn2,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.6.0 -neuronx-py310-sdk2.23 .0-ubuntu22.04 | torchserve 0.11.0 |
1818| [ PyTorch 2.1.2] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/inference/2.1.2/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.20.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.20.2-ubuntu20.04 | torchserve 0.11.0 |
1919| [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/inference/1.13.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.20.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.20.2-ubuntu20.04 | torchserve 0.11.0 |
2020
2121### pytorch-training-neuronx
2222
2323| Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL |
2424| ----------------------------------------------------------------------------------------------------------------------------------------| -------------------------------------------------------| --------------------| ------------------------------| ------------------------| -------------------------------------------------------------------------------------------|
25- | [ PyTorch 2.5.1 ] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.22 .0/docker/pytorch/training/2.5.1 /Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_training, torch-neuronx | Neuron 2.22 .0 | trn1,trn2,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.5.1 -neuronx-py310-sdk2.22 .0-ubuntu22.04 |
25+ | [ PyTorch 2.6.0 ] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.23 .0/docker/pytorch/training/2.6.0 /Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_training, torch-neuronx | Neuron 2.23 .0 | trn1,trn2,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.6.0 -neuronx-py310-sdk2.23 .0-ubuntu22.04 |
2626| [ PyTorch 2.1.2] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/training/2.1.2/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_training, torch-neuronx | Neuron 2.20.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.20.2-ubuntu20.04 |
2727| [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/training/1.13.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_training, torch-neuronx | Neuron 2.20.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.20.2-ubuntu20.04 |
2828
2929### jax-training-neuronx
3030
3131| Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | Other Packages |
3232| ----------------------------------------------------------------------------------------------------------------------------------------| ---------------------------------| --------------------| ------------------------------| ------------------------| ------------------------------------------------------------------------------------------| -------------------|
33- | [ JAX 0.5] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.22 .0/docker/jax/training/0.5/Dockerfile.neuronx ) | jax-neuronx, libneuronxla | Neuron 2.22 .0 | trn1,trn2,inf2 | 3.10 (py310) | public.ecr.aws/neuron/jax-training-neuronx:0.5-neuronx-py310-sdk2.22 .0-ubuntu22.04 | jaxlib 0.5 |
33+ | [ JAX 0.5] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.23 .0/docker/jax/training/0.5/Dockerfile.neuronx ) | jax-neuronx, libneuronxla | Neuron 2.23 .0 | trn1,trn2,inf2 | 3.10 (py310) | public.ecr.aws/neuron/jax-training-neuronx:0.5-neuronx-py310-sdk2.23 .0-ubuntu22.04 | jaxlib 0.5 |
3434
3535## Security
3636
0 commit comments