From 2076bfddc4017c7f98ec2d9da8523cf8943db456 Mon Sep 17 00:00:00 2001 From: Bob Han Date: Wed, 20 Nov 2024 18:32:44 -0800 Subject: [PATCH] Add Neuron(X) SDK 2.20.2 images to release_images.md --- available_images.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/available_images.md b/available_images.md index dce9a89ca8b2..ea77c6777bfd 100644 --- a/available_images.md +++ b/available_images.md @@ -289,11 +289,11 @@ Note: Starting from Neuron SDK 2.17.0, Dockerfiles for PyTorch Neuron Containers | Framework | Neuron Package | Neuron SDK Version | Job Type | Supported EC2 Instance Types | Python Version Options | Example URL | |----------------------------------------------------------------------------------------------------------------------------------------------------|----------------------------------------------------------|--------------------|-----------|------------------------------|------------------------|----------------------------------------------------------------------------------------------------------------------| -| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.1/docker/pytorch/inference/2.1.2/Dockerfile.neuronx) | torch-neuronx, transformers-neuronx, neuronx_distributed | Neuron 2.20.1 | inference | trn1,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.20.1-ubuntu20.04 | -| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.1/docker/pytorch/training/2.1.2/Dockerfile.neuronx) | torch-neuronx, neuronx_distributed | Neuron 2.20.1 | training | trn1, inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.20.1-ubuntu20.04 | -| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.1/docker/pytorch/inference/1.13.1/Dockerfile.neuron) | torch-neuron | Neuron 2.20.1 | inference | inf1 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuron:1.13.1-neuron-py310-sdk2.20.1-ubuntu20.04 | -| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.1/docker/pytorch/inference/1.13.1/Dockerfile.neuronx) | torch-neuronx, transformers-neuronx, neuronx_distributed | Neuron 2.20.1 | inference | trn1,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.20.1-ubuntu20.04 | -| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.1/docker/pytorch/training/1.13.1/Dockerfile.neuronx) | torch-neuronx, neuronx_distributed | Neuron 2.20.1 | training | trn1, inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.20.1-ubuntu20.04 | +| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/inference/2.1.2/Dockerfile.neuronx) | torch-neuronx, transformers-neuronx, neuronx_distributed | Neuron 2.20.2 | inference | trn1,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.20.2-ubuntu20.04 | +| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/training/2.1.2/Dockerfile.neuronx) | torch-neuronx, neuronx_distributed | Neuron 2.20.2 | training | trn1, inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.20.2-ubuntu20.04 | +| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/inference/1.13.1/Dockerfile.neuron) | torch-neuron | Neuron 2.20.2 | inference | inf1 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuron:1.13.1-neuron-py310-sdk2.20.2-ubuntu20.04 | +| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/inference/1.13.1/Dockerfile.neuronx) | torch-neuronx, transformers-neuronx, neuronx_distributed | Neuron 2.20.2 | inference | trn1,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.20.2-ubuntu20.04 | +| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.2/docker/pytorch/training/1.13.1/Dockerfile.neuronx) | torch-neuronx, neuronx_distributed | Neuron 2.20.2 | training | trn1, inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.20.2-ubuntu20.04 | | [Tensorflow 2.10.1](https://github.com/aws/deep-learning-containers/blob/master/tensorflow/inference/docker/2.10/py3/sdk2.17.0/Dockerfile.neuron) | tensorflow-neuron | Neuron 2.17.0 | inference | inf1 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference-neuron:2.10.1-neuron-py310-sdk2.17.0-ubuntu20.04 | | [Tensorflow 2.10.1](https://github.com/aws/deep-learning-containers/blob/master/tensorflow/inference/docker/2.10/py3/sdk2.17.0/Dockerfile.neuronx) | tensorflow-neuronx | Neuron 2.17.0 | inference | trn1,inf2 | 3.10 (py310) | 763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference-neuronx:2.10.1-neuronx-py310-sdk2.17.0-ubuntu20.04 |