@@ -8,21 +8,21 @@ AWS Neuron Deep Learning Containers (DLCs) are a set of Docker images for traini
8
8
9
9
| Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | Other Packages |
10
10
| ----------------------------------------------------------------------------------------------------------------------------------------| ---------------------------------| --------------------| ------------------------------| ------------------------| ------------------------------------------------------------------------------------------| -------------------|
11
- | [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.18.2 /docker/pytorch/inference/1.13.1/Dockerfile.neuron ) | aws-neuronx-tools, torch-neuron | Neuron 2.18.2 | inf1 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuron:1.13.1-neuron-py310-sdk2.18.2 -ubuntu20.04 | torchserve 0.10 .0 |
11
+ | [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.19.0 /docker/pytorch/inference/1.13.1/Dockerfile.neuron ) | aws-neuronx-tools, torch-neuron | Neuron 2.19.0 | inf1 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuron:1.13.1-neuron-py310-sdk2.19.0 -ubuntu20.04 | torchserve 0.11 .0 |
12
12
13
13
### pytorch-inference-neuronx
14
14
15
15
| Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL | Other Packages |
16
16
| -----------------------------------------------------------------------------------------------------------------------------------------| -----------------------------------------------------------------------------| --------------------| ------------------------------| ------------------------| --------------------------------------------------------------------------------------------| -------------------|
17
- | [ PyTorch 2.1.2] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.18.2 /docker/pytorch/inference/2.1.2/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.18.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.18.2 -ubuntu20.04 | torchserve 0.10 .0 |
18
- | [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.18.2 /docker/pytorch/inference/1.13.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.18.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.18.2 -ubuntu20.04 | torchserve 0.10 .0 |
17
+ | [ PyTorch 2.1.2] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.19.0 /docker/pytorch/inference/2.1.2/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.19.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:2.1.2-neuronx-py310-sdk2.19.0 -ubuntu20.04 | torchserve 0.11 .0 |
18
+ | [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.19.0 /docker/pytorch/inference/1.13.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx, transformers-neuronx | Neuron 2.19.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.19.0 -ubuntu20.04 | torchserve 0.11 .0 |
19
19
20
20
### pytorch-training-neuronx
21
21
22
22
| Framework | Neuron Packages | Neuron SDK Version | Supported EC2 Instance Types | Python Version Options | ECR Public URL |
23
23
| ----------------------------------------------------------------------------------------------------------------------------------------| -------------------------------------------------------| --------------------| ------------------------------| ------------------------| -------------------------------------------------------------------------------------------|
24
- | [ PyTorch 2.1.2] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.18.2 /docker/pytorch/training/2.1.2/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx | Neuron 2.18.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.18.2 -ubuntu20.04 |
25
- | [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.18.2 /docker/pytorch/training/1.13.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx | Neuron 2.18.2 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.18.2 -ubuntu20.04 |
24
+ | [ PyTorch 2.1.2] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.19.0 /docker/pytorch/training/2.1.2/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx | Neuron 2.19.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:2.1.2-neuronx-py310-sdk2.19.0 -ubuntu20.04 |
25
+ | [ PyTorch 1.13.1] ( https://github.com/aws-neuron/deep-learning-containers/blob/2.19.0 /docker/pytorch/training/1.13.1/Dockerfile.neuronx ) | aws-neuronx-tools, neuronx_distributed, torch-neuronx | Neuron 2.19.0 | trn1,inf2 | 3.10 (py310) | public.ecr.aws/neuron/pytorch-training-neuronx:1.13.1-neuronx-py310-sdk2.19.0 -ubuntu20.04 |
26
26
27
27
## Security
28
28
0 commit comments