@@ -352,8 +352,8 @@ Note: Starting from Neuron SDK 2.17.0, Dockerfiles for PyTorch Neuron Containers
352352
353353| Framework | Neuron Package | Neuron SDK Version | Job Type | Supported EC2 Instance Types | Python Version Options | Example URL |
354354| ----------------------------------------------------------------------------------------------------------------------------------------------------| ----------------------------------------------------------| --------------------| -----------| ------------------------------| ------------------------| ----------------------------------------------------------------------------------------------------------------------|
355- | [ PyTorch 2.8.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.26.0/docker/ pytorch/inference/2.8.0/Dockerfile.neuronx ) | torch-neuronx, neuronx_distributed, neuronx_distributed_inference | Neuron 2.26.0 | inference | trn1,trn2,inf2 | 3.11 (py311) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:2.8.0-neuronx-py311-sdk2.26.0 -ubuntu22.04 |
356- | [ PyTorch 2.8.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.26.0/docker/ pytorch/training/2.8.0/Dockerfile.neuronx ) | torch-neuronx, neuronx_distributed, neuronx_distributed_training | Neuron 2.26.0 | training | trn1,trn2,inf2 | 3.11 (py311) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:2.8.0-neuronx-py311-sdk2.26.0 -ubuntu22.04 |
355+ | [ PyTorch 2.8.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.26.1/ pytorch/inference/2.8.0/Dockerfile.neuronx ) | torch-neuronx, neuronx_distributed, neuronx_distributed_inference | Neuron 2.26.1 | inference | trn1,trn2,inf2 | 3.11 (py311) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:2.8.0-neuronx-py311-sdk2.26.1 -ubuntu22.04 |
356+ | [ PyTorch 2.8.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.26.1/ pytorch/training/2.8.0/Dockerfile.neuronx ) | torch-neuronx, neuronx_distributed, neuronx_distributed_training | Neuron 2.26.1 | training | trn1,trn2,inf2 | 3.11 (py311) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:2.8.0-neuronx-py311-sdk2.26.1 -ubuntu22.04 |
357357| [ PyTorch 2.7.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.25.0/docker/pytorch/inference/2.7.0/Dockerfile.neuronx ) | torch-neuronx, transformers-neuronx, neuronx_distributed, neuronx_distributed_inference | Neuron 2.25.0 | inference | trn1,trn2,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04 |
358358| [ PyTorch 2.7.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.25.0/docker/pytorch/training/2.7.0/Dockerfile.neuronx ) | torch-neuronx, transformers-neuronx, neuronx_distributed, neuronx_distributed_training | Neuron 2.25.0 | training | trn1,trn2,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04 |
359359| [ PyTorch 2.7.0] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.24.1/docker/pytorch/inference/2.7.0/Dockerfile.neuronx ) | torch-neuronx, transformers-neuronx, neuronx_distributed, neuronx_distributed_inference | Neuron 2.24.1 | inference | trn1,trn2,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:2.7.0-neuronx-py310-sdk2.24.1-ubuntu22.04 |
0 commit comments