Skip to content

Commit

Permalink
Add PT 2.3 Graviton to available_images (#4037)
Browse files Browse the repository at this point in the history
  • Loading branch information
@sallyseok
sallyseok authored Jul 4, 2024
1 parent a17f249 commit b3891c8
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions available_images.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -129,15 +129,15 @@ EC2 Framework Graviton Containers (EC2, ECS, and EKS support only)

| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options | Example URL |
|-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------------------|
| PyTorch 2.2.1 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.2.1-cpu-py310-ubuntu20.04-ec2 |
| PyTorch 2.3.0 |inference |No |CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.3.0-cpu-py311-ubuntu20.04-ec2 |
| TensorFlow 2.14.1 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-graviton:2.14.1-cpu-py310-ubuntu20.04-ec2 |

SageMaker Framework Graviton Containers (SM support only)
============================

| Framework | Job Type | Horovod Options| CPU/GPU | Python Version Options | Example URL |
|-------------------|-----------|----------------|---------|------------------------|-------------------------------------------------------------------------------------------------------------------|
| PyTorch 2.2.1 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.2.1-cpu-py310-ubuntu20.04-sagemaker |
| PyTorch 2.3.0 | inference | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.3.0-cpu-py311-ubuntu20.04-sagemaker |
| TensorFlow 2.14.1 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-graviton:2.14.1-cpu-py310-ubuntu20.04-sagemaker |

NVIDIA Triton Inference Containers (SM support only)
Expand Down Expand Up @@ -349,13 +349,15 @@ Prior EC2 Framework Graviton Containers

| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options |Example URL |
|-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|
| PyTorch 2.2.1 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.2.1-cpu-py310-ubuntu20.04-ec2 |
| PyTorch 2.1.0 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.1.0-cpu-py310-ubuntu20.04-ec2
| TensorFlow 2.13.0 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-graviton:2.13.0-cpu-py310-ubuntu20.04-ec2 |

Prior SageMaker Framework Graviton Containers
============================
| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options |Example URL |
|-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|
| PyTorch 2.2.1 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.2.1-cpu-py310-ubuntu20.04-sagemaker |
| PyTorch 2.1.0 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.1.0-cpu-py310-ubuntu20.04-sagemaker |
| TensorFlow 2.13.0 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-graviton:2.13.0-cpu-py310-ubuntu20.04-sagemaker |

Expand Down

0 comments on commit b3891c8

Please sign in to comment.