PT 2.2 end of support (#4530)

aws · Feb 4, 2025 · 9c9977b · 9c9977b
1 parent ba9f23b
commit 9c9977b
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 109 deletions.
diff --git a/available_images.md b/available_images.md
@@ -329,10 +329,6 @@ Prior EC2 Framework Container Versions
 | PyTorch 2.3.0     |training	|No			|GPU 		| 3.11 (py311)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.3.0-gpu-py311-cu121-ubuntu20.04-ec2       |
 | PyTorch 2.3.0     |inference	|No			|CPU 		| 3.11 (py311)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.3.0-cpu-py311-ubuntu20.04-ec2            |
 | PyTorch 2.3.0     |inference	|No			|GPU 		| 3.11 (py311)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.3.0-gpu-py311-cu121-ubuntu20.04-ec2      |
-| PyTorch 2.2.0     |training	|No			|CPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.2.0-cpu-py310-ubuntu20.04-ec2             |
-| PyTorch 2.2.0     |training	|No			|GPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-ec2       |
-| PyTorch 2.2.0     |inference	|No			|CPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.2.0-cpu-py310-ubuntu20.04-ec2            |
-| PyTorch 2.2.0     |inference	|No			|GPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.2.0-gpu-py310-cu118-ubuntu20.04-ec2      |
 | TensorFlow 2.16.2 |training	|No			|CPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.16.2-cpu-py310-ubuntu20.04-ec2		      |
 | TensorFlow 2.16.2 |training	|No			|GPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.16.2-gpu-py310-cu123-ubuntu20.04-ec2	 |
 | TensorFlow 2.16.1 |inference	|No			|CPU 		| 3.10 (py310)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.16.1-cpu-py310-ubuntu20.04-ec2		      |
@@ -350,10 +346,6 @@ Prior SageMaker Framework Container Versions
 | PyTorch 2.3.0     | training	| No			  | GPU 	  | 3.11 (py311)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker     |
 | PyTorch 2.3.0     | inference	| No			  | CPU 	  | 3.11 (py311)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.3.0-cpu-py311-ubuntu20.04-sagemaker           |
 | PyTorch 2.3.0     | inference	| No			  | GPU 	  | 3.11 (py311)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker     |
-| PyTorch 2.2.0     | training   | No 			  | CPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.2.0-cpu-py310-ubuntu20.04-sagemaker            |
-| PyTorch 2.2.0     | training   | No			  | GPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker      |
-| PyTorch 2.2.0     | inference	| No			  | CPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.2.0-cpu-py310-ubuntu20.04-sagemaker           |
-| PyTorch 2.2.0     | inference	| No			  | GPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.2.0-gpu-py310-cu118-ubuntu20.04-sagemaker     |
 | TensorFlow 2.16.2 | training  | No			  | CPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.16.2-cpu-py310-ubuntu20.04-sagemaker		      |
 | TensorFlow 2.16.2 | training  | No			  | GPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.16.2-gpu-py310-cu123-ubuntu20.04-sagemaker  |
 | TensorFlow 2.16.1 | inference | No			  | CPU 	  | 3.10 (py310)			   | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.16.1-cpu-py310-ubuntu20.04-sagemaker	      |
@@ -366,15 +358,13 @@ Prior EC2 Framework ARM64/Graviton Containers
 | PyTorch 2.4.0     |inference	|No			|CPU 		| 3.11 (py311)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.4.0-cpu-py311-ubuntu22.04-ec2		     |
 | PyTorch 2.4.0     |inference	|No			|GPU 		| 3.11 (py311)			| 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.4.0-gpu-py311-cu124-ubuntu22.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		     |
-| 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 |
 
 Prior SageMaker Framework ARM64/Graviton Containers
 ============================
 | Framework         |Job Type	|Horovod Options|CPU/GPU 	|Python Version Options	|Example URL																						|
 |-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|
 | PyTorch 2.4.0     | inference | No			 | CPU 	   | 3.11 (py311)			    | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-graviton:2.4.0-cpu-py311-ubuntu22.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     |
-| 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     |
 
 Prior AutoGluon Training Containers
 ===============================
@@ -395,7 +385,7 @@ Prior AutoGluon Inference Containers
 Prior SageMaker Training Compiler Containers
 ===============================
 
-| Framework                                     |Job Type	|CPU/GPU 	|Python Version Options	|Example URL																						|
+| Framework                                    |Job Type	|CPU/GPU 	|Python Version Options	|Example URL																						|
 |-----------------------------------------------|-----------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|
 |TensorFlow 2.6.3 with HuggingFace transformers 4.17.0 and SageMaker Training Compiler |training	|GPU 		| 3.8 (py38)			|763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-tensorflow-trcomp-training:2.6.3-transformers4.17.0-gpu-py38-cu112-ubuntu20.04 	|
 |PyTorch 1.12.0 with SageMaker Training Compiler    |training	|GPU 		| 3.8 (py38)			|763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-trcomp-training:1.12.0-gpu-py38-cu113-ubuntu20.04-sagemaker     |

diff --git a/release_images_patches.yml b/release_images_patches.yml
@@ -1,86 +1,8 @@
 ---
 # DLC images to patch - force_release is not allowed in this file
 release_images:
-  # PyTorch 2.2 Images
-  1:
-    framework: "pytorch"
-    version: "2.2.0"
-    arch_type: "x86"
-    customer_type: "ec2"
-    training:
-      device_types: [ "cpu", "gpu" ]
-      python_versions: [ "py310" ]
-      os_version: "ubuntu20.04"
-      cuda_version: "cu121"
-      example: False
-      disable_sm_tag: False
-      force_release: False
-  2:
-    framework: "pytorch"
-    version: "2.2.0"
-    arch_type: "x86"
-    customer_type: "sagemaker"
-    training:
-      device_types: [ "cpu", "gpu" ]
-      python_versions: [ "py310" ]
-      os_version: "ubuntu20.04"
-      cuda_version: "cu121"
-      example: False
-      disable_sm_tag: False
-      force_release: False
-  3:
-    framework: "pytorch"
-    version: "2.2.0"
-    arch_type: "x86"
-    customer_type: "ec2"
-    inference:
-      device_types: [ "cpu", "gpu" ]
-      python_versions: [ "py310" ]
-      os_version: "ubuntu20.04"
-      cuda_version: "cu118"
-      example: False
-      disable_sm_tag: False
-      force_release: False
-  4:
-    framework: "pytorch"
-    version: "2.2.0"
-    arch_type: "x86"
-    customer_type: "sagemaker"
-    inference:
-      device_types: [ "cpu", "gpu" ]
-      python_versions: [ "py310" ]
-      os_version: "ubuntu20.04"
-      cuda_version: "cu118"
-      example: False
-      disable_sm_tag: False
-      force_release: False
-  # 5:
-  #   framework: "pytorch"
-  #   version: "2.2.1"
-  #   arch_type: "graviton"
-  #   customer_type: "ec2"
-  #   inference:
-  #     device_types: [ "cpu" ]
-  #     python_versions: [ "py310" ]
-  #     os_version: "ubuntu20.04"
-  #     example: False
-  #     disable_sm_tag: False
-  #     force_release: False
-  # 6:
-  #   framework: "pytorch"
-  #   version: "2.2.1"
-  #   arch_type: "graviton"
-  #   customer_type: "sagemaker"
-  #   inference:
-  #     device_types: [ "cpu" ]
-  #     python_versions: [ "py310" ]
-  #     os_version: "ubuntu20.04"
-  #     example: False
-  #     disable_sm_tag: False
-  #     force_release: False
-
   # PyTorch 2.3 Images
-  5:
+  1:
     framework: "pytorch"
     version: "2.3.0"
     customer_type: "ec2"
@@ -93,7 +15,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  6:
+  2:
     framework: "pytorch"
     version: "2.3.0"
     customer_type: "sagemaker"
@@ -106,7 +28,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  7:
+  3:
     framework: "pytorch"
     version: "2.3.0"
     arch_type: "x86"
@@ -119,7 +41,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  8:
+  4:
     framework: "pytorch"
     version: "2.3.0"
     arch_type: "x86"
@@ -158,7 +80,7 @@ release_images:
   #     force_release: False
 
   # PyTorch 2.4 Images
-  9:
+  5:
     framework: "pytorch"
     version: "2.4.0"
     customer_type: "ec2"
@@ -171,7 +93,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  10:
+  6:
     framework: "pytorch"
     version: "2.4.0"
     customer_type: "sagemaker"
@@ -184,7 +106,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  11:
+  7:
     framework: "pytorch"
     version: "2.4.0"
     arch_type: "x86"
@@ -197,7 +119,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  12:
+  8:
     framework: "pytorch"
     version: "2.4.0"
     arch_type: "x86"
@@ -237,7 +159,7 @@ release_images:
   #     force_release: False
 
   # PyTorch 2.5 Images
-  13:
+  9:
     framework: "pytorch"
     version: "2.5.1"
     customer_type: "ec2"
@@ -250,7 +172,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  14:
+  10:
     framework: "pytorch"
     version: "2.5.1"
     customer_type: "sagemaker"
@@ -263,7 +185,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  15:
+  11:
     framework: "pytorch"
     version: "2.5.1"
     arch_type: "x86"
@@ -276,7 +198,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  16:
+  12:
     framework: "pytorch"
     version: "2.5.1"
     arch_type: "x86"
@@ -316,7 +238,7 @@ release_images:
   #     force_release: False
 
   # TensorFlow 2.16 Images
-  17:
+  13:
     framework: "tensorflow"
     version: "2.16.2"
     customer_type: "ec2"
@@ -329,7 +251,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  18:
+  14:
     framework: "tensorflow"
     version: "2.16.2"
     customer_type: "sagemaker"
@@ -342,7 +264,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  19:
+  15:
     framework: "tensorflow"
     version: "2.16.1"
     arch_type: "x86"
@@ -355,7 +277,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  20:
+  16:
     framework: "tensorflow"
     version: "2.16.1"
     arch_type: "x86"
@@ -370,7 +292,7 @@ release_images:
       force_release: False
 
   # TensorFlow 2.18 Images
-  21:
+  17:
     framework: "tensorflow"
     version: "2.18.0"
     customer_type: "ec2"
@@ -383,7 +305,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  22:
+  18:
     framework: "tensorflow"
     version: "2.18.0"
     customer_type: "sagemaker"
@@ -396,7 +318,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  23:
+  19:
     framework: "tensorflow"
     version: "2.18.0"
     arch_type: "x86"
@@ -409,7 +331,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  24:
+  20:
     framework: "tensorflow"
     version: "2.18.0"
     arch_type: "x86"