From 4217d71bfb68819a45f3b772a6818084dd438cee Mon Sep 17 00:00:00 2001 From: geetasg Date: Tue, 17 Sep 2024 16:19:57 -0700 Subject: [PATCH] Update Neuron DLCs to 2.20.0 in README (#20) *Issue #, if available:* *Description of changes:* Update Neuron DLCs to 2.20.0 in README By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. *Issue #, if available:* *Description of changes:* By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. Signed-off-by: Geeta Gharpure Co-authored-by: Geeta Gharpure --- README.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 47587ff..ede528b 100644 --- a/README.md +++ b/README.md @@ -8,21 +8,21 @@ AWS Neuron Deep Learning Containers (DLCs) are a set of Docker images for traini | Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | Other Packages | |----------------------------------------------------------------------------------------------------------------------------------------|---------------------------------|--------------------|------------------------------|------------------------|------------------------------------------------------------------------------------------|-------------------| -| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.19.1/docker/pytorch/inference/1.13.1/Dockerfile.neuron) | aws-neuronx-tools, torch-neuron | Neuron 2.19.1 | inf1 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuron:1.13.1-neuron-py310-sdk2.19.1-ubuntu20.04 | torchserve 0.11.0 | +| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.0/docker/pytorch/inference/1.13.1/Dockerfile.neuron) | aws-neuronx-tools, torch-neuron | Neuron 2.20.0 | inf1 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuron:1.13.1-neuron-py310-sdk2.20.0-ubuntu20.04 | torchserve 0.11.0 | ### pytorch-inference-neuronx | Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | Other Packages | |-----------------------------------------------------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------|--------------------|------------------------------|------------------------|--------------------------------------------------------------------------------------------|-------------------| -| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.19.1/docker/pytorch/inference/2.1.2/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.19.1 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.19.1-ubuntu20.04 | torchserve 0.11.0 | -| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.19.1/docker/pytorch/inference/1.13.1/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.19.1 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.19.1-ubuntu20.04 | torchserve 0.11.0 | +| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.0/docker/pytorch/inference/2.1.2/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.20.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.20.0-ubuntu20.04 | torchserve 0.11.0 | +| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.0/docker/pytorch/inference/1.13.1/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.20.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.20.0-ubuntu20.04 | torchserve 0.11.0 | ### pytorch-training-neuronx | Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | |----------------------------------------------------------------------------------------------------------------------------------------|-------------------------------------------------------|--------------------|------------------------------|------------------------|-------------------------------------------------------------------------------------------| -| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.19.1/docker/pytorch/training/2.1.2/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, torch-neuronx | Neuron 2.19.1 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.19.1-ubuntu20.04 | -| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.19.1/docker/pytorch/training/1.13.1/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, torch-neuronx | Neuron 2.19.1 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.19.1-ubuntu20.04 | +| [PyTorch 2.1.2](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.0/docker/pytorch/training/2.1.2/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_training, torch-neuronx | Neuron 2.20.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.20.0-ubuntu20.04 | +| [PyTorch 1.13.1](https://github.com/aws-neuron/deep-learning-containers/blob/2.20.0/docker/pytorch/training/1.13.1/Dockerfile.neuronx) | aws-neuronx-tools, neuronx_distributed, neuronx_distributed_training, torch-neuronx | Neuron 2.20.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.20.0-ubuntu20.04 | ## Security