Skip to content

Commit d8a8160

Browse files
DevakiBolleneniDevakiBolleneni
andauthored
Deprecate TensorFlow 2.18 (#5413)
* Deprecate TensorFlow 2.18 --------- Co-authored-by: DevakiBolleneni <devakib@amazon.com>
1 parent f58575e commit d8a8160

File tree

2 files changed

+3
-101
lines changed

2 files changed

+3
-101
lines changed

available_images.md

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,6 @@ EC2 Framework Containers (EC2, ECS, and EKS support only)
121121
| PyTorch 2.8.0 |training |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.8.0-gpu-py312-cu129-ubuntu22.04-ec2 |
122122
| PyTorch 2.6.0 |inference |No |CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.6.0-cpu-py312-ubuntu22.04-ec2 |
123123
| PyTorch 2.6.0 |inference |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.6.0-gpu-py312-cu124-ubuntu22.04-ec2 |
124-
| TensorFlow 2.18.0 |training |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.18.0-cpu-py310-ubuntu22.04-ec2 |
125-
| TensorFlow 2.18.0 |training |No |GPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.18.0-gpu-py310-cu125-ubuntu22.04-ec2 |
126-
| TensorFlow 2.18.0 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.18.0-cpu-py310-ubuntu20.04-ec2 |
127-
| TensorFlow 2.18.0 |inference |No |GPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.18.0-gpu-py310-cu122-ubuntu20.04-ec2 |
128-
129124

130125
SageMaker Framework Containers (SM support only)
131126
============================
@@ -152,8 +147,6 @@ instead of "pytorch-inference-graviton". Graviton DLCs and ARM64 DLCs are functi
152147
| PyTorch 2.7.0 |training |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training-arm64:2.7.0-gpu-py312-cu128-ubuntu22.04-ec2 |
153148
| PyTorch 2.6.0 |inference |No |CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-arm64:2.6.0-cpu-py312-ubuntu22.04-ec2 |
154149
| PyTorch 2.6.0 |inference |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-arm64:2.6.0-gpu-py312-cu124-ubuntu22.04-ec2 |
155-
| TensorFlow 2.18.0 |inference |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-arm64:2.18.0-cpu-py310-ubuntu20.04-ec2 |
156-
157150

158151
SageMaker Framework ARM64/Graviton Containers (SM support only)
159152
============================
@@ -163,8 +156,6 @@ instead of "pytorch-inference-graviton". Graviton DLCs and ARM64 DLCs are functi
163156
| Framework | Job Type | Horovod Options| CPU/GPU | Python Version Options | Example URL |
164157
|-------------------|-----------|----------------|---------|------------------------|-------------------------------------------------------------------------------------------------------------------|
165158
| PyTorch 2.6.0 | inference | No | CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-arm64:2.6.0-cpu-py312-ubuntu22.04-sagemaker |
166-
| TensorFlow 2.18.0 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-arm64:2.18.0-cpu-py310-ubuntu20.04-sagemaker |
167-
168159

169160
NVIDIA Triton Inference Containers (SM support only)
170161
============================
@@ -412,10 +403,6 @@ Prior SageMaker Framework Container Versions
412403
| PyTorch 2.5.1 | training | No | GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-gpu-py311-cu124-ubuntu22.04-sagemaker |
413404
| PyTorch 2.5.1 | inference | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.5.1-cpu-py311-ubuntu22.04-sagemaker |
414405
| PyTorch 2.5.1 | inference | No | GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.5.1-gpu-py311-cu124-ubuntu22.04-sagemaker |
415-
| TensorFlow 2.18.0 | training | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.18.0-cpu-py310-ubuntu22.04-sagemaker |
416-
| TensorFlow 2.18.0 | training | No | GPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.18.0-gpu-py310-cu125-ubuntu22.04-sagemaker |
417-
| TensorFlow 2.18.0 | inference | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.18.0-cpu-py310-ubuntu20.04-sagemaker |
418-
| TensorFlow 2.18.0 | inference | No | GPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.18.0-gpu-py310-cu122-ubuntu20.04-sagemaker |
419406

420407
Prior EC2 Framework ARM64/Graviton Containers
421408
============================

release_images_patches.yml

Lines changed: 3 additions & 88 deletions
Original file line numberDiff line numberDiff line change
@@ -250,93 +250,8 @@ release_images:
250250
public_registry: True
251251
enable_soci: True
252252

253-
# TensorFlow 2.18 Images
254-
18:
255-
framework: "tensorflow"
256-
version: "2.18.0"
257-
customer_type: "ec2"
258-
arch_type: "x86"
259-
training:
260-
device_types: [ "cpu", "gpu" ]
261-
python_versions: [ "py310" ]
262-
os_version: "ubuntu22.04"
263-
cuda_version: "cu125"
264-
example: False
265-
disable_sm_tag: False
266-
force_release: False
267-
public_registry: True
268-
19:
269-
framework: "tensorflow"
270-
version: "2.18.0"
271-
customer_type: "sagemaker"
272-
arch_type: "x86"
273-
training:
274-
device_types: [ "cpu", "gpu" ]
275-
python_versions: [ "py310" ]
276-
os_version: "ubuntu22.04"
277-
cuda_version: "cu125"
278-
example: False
279-
disable_sm_tag: False
280-
force_release: False
281-
public_registry: True
282-
20:
283-
framework: "tensorflow"
284-
version: "2.18.0"
285-
arch_type: "x86"
286-
customer_type: "ec2"
287-
inference:
288-
device_types: [ "cpu", "gpu" ]
289-
python_versions: [ "py310" ]
290-
os_version: "ubuntu20.04"
291-
cuda_version: "cu122"
292-
example: False
293-
disable_sm_tag: False
294-
force_release: False
295-
public_registry: True
296-
21:
297-
framework: "tensorflow"
298-
version: "2.18.0"
299-
arch_type: "x86"
300-
customer_type: "sagemaker"
301-
inference:
302-
device_types: [ "cpu", "gpu" ]
303-
python_versions: [ "py310" ]
304-
os_version: "ubuntu20.04"
305-
cuda_version: "cu122"
306-
example: False
307-
disable_sm_tag: False
308-
force_release: False
309-
public_registry: True
310-
22:
311-
framework: "tensorflow"
312-
version: "2.18.0"
313-
arch_type: "arm64"
314-
customer_type: "ec2"
315-
inference:
316-
device_types: [ "cpu" ]
317-
python_versions: [ "py310" ]
318-
os_version: "ubuntu20.04"
319-
cuda_version: "cu122"
320-
example: False
321-
disable_sm_tag: False
322-
force_release: False
323-
public_registry: True
324-
23:
325-
framework: "tensorflow"
326-
version: "2.18.0"
327-
arch_type: "arm64"
328-
customer_type: "sagemaker"
329-
inference:
330-
device_types: [ "cpu" ]
331-
python_versions: [ "py310" ]
332-
os_version: "ubuntu20.04"
333-
cuda_version: "cu122"
334-
example: False
335-
disable_sm_tag: False
336-
force_release: False
337-
public_registry: True
338253
# TensorFlow 2.19 Images
339-
24:
254+
18:
340255
framework: "tensorflow"
341256
version: "2.19.0"
342257
customer_type: "sagemaker"
@@ -351,7 +266,7 @@ release_images:
351266
force_release: False
352267
public_registry: True
353268
enable_soci: True
354-
25:
269+
19:
355270
framework: "tensorflow"
356271
version: "2.19.0"
357272
arch_type: "x86"
@@ -365,7 +280,7 @@ release_images:
365280
disable_sm_tag: False
366281
force_release: False
367282
public_registry: True
368-
26:
283+
20:
369284
framework: "tensorflow"
370285
version: "2.19.0"
371286
arch_type: "arm64"

0 commit comments

Comments
 (0)