diff --git a/pytorch/training/buildspec-2-5-ec2.yml b/pytorch/training/buildspec-2-5-ec2.yml
index f79557bbdf12..9b1dd48f38d0 100644
--- a/pytorch/training/buildspec-2-5-ec2.yml
+++ b/pytorch/training/buildspec-2-5-ec2.yml
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
 version: &VERSION 2.5.1
 short_version: &SHORT_VERSION "2.5"
 arch_type: x86
-# autopatch_build: "True"
+autopatch_build: "True"
 
 repository_info:
   training_repository: &TRAINING_REPOSITORY
diff --git a/pytorch/training/buildspec-2-5-sm.yml b/pytorch/training/buildspec-2-5-sm.yml
index dbf3fb1db102..1617c86518da 100644
--- a/pytorch/training/buildspec-2-5-sm.yml
+++ b/pytorch/training/buildspec-2-5-sm.yml
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
 version: &VERSION 2.5.1
 short_version: &SHORT_VERSION "2.5"
 arch_type: x86
-# autopatch_build: "True"
+autopatch_build: "True"
 
 repository_info:
   training_repository: &TRAINING_REPOSITORY
diff --git a/pytorch/training/docker/2.5/py3/Dockerfile.ec2.cpu.core_packages.json b/pytorch/training/docker/2.5/py3/Dockerfile.ec2.cpu.core_packages.json
new file mode 100644
index 000000000000..35db7240638c
--- /dev/null
+++ b/pytorch/training/docker/2.5/py3/Dockerfile.ec2.cpu.core_packages.json
@@ -0,0 +1,51 @@
+{
+  "accelerate": {
+    "version_specifier": "==1.1.1",
+    "skip": "True"
+  },
+  "fastai": {
+    "version_specifier": "==2.7.18",
+    "skip": "True"
+  },
+  "s3torchconnector": {
+    "version_specifier": "==1.2.6",
+    "skip": "True"
+  },
+  "torchaudio": {
+    "version_specifier": "==2.5.1+cpu",
+    "skip": "True"
+  },
+  "torchtext": {
+    "version_specifier": "==0.18.0+cpu",
+    "skip": "True"
+  },
+  "torchtnt": {
+    "version_specifier": "==0.2.4",
+    "skip": "True"
+  },
+  "torchvision": {
+    "version_specifier": "==0.20.1+cpu",
+    "skip": "True"
+  },
+  "idna": {
+    "version_specifier": ">=3.7"
+  },
+  "tqdm": {
+    "version_specifier": ">=4.66.3"
+  },
+  "requests": {
+    "version_specifier": ">=2.32.0"
+  },
+  "setuptools": {
+    "version_specifier": ">=70.0.0"
+  },
+  "urllib3": {
+    "version_specifier": "<2"
+  },
+  "awscli": {
+    "version_specifier": "<2"
+  },
+  "numpy": {
+    "version_specifier": "<2"
+  }
+}
diff --git a/pytorch/training/docker/2.5/py3/Dockerfile.sagemaker.cpu.core_packages.json b/pytorch/training/docker/2.5/py3/Dockerfile.sagemaker.cpu.core_packages.json
new file mode 100644
index 000000000000..e68cc40a46ca
--- /dev/null
+++ b/pytorch/training/docker/2.5/py3/Dockerfile.sagemaker.cpu.core_packages.json
@@ -0,0 +1,60 @@
+{
+  "accelerate": {
+    "version_specifier": "==1.1.1",
+    "skip": "True"
+  },
+  "fastai": {
+    "version_specifier": "==2.7.18",
+    "skip": "True"
+  },
+  "s3torchconnector": {
+    "version_specifier": "==1.2.6",
+    "skip": "True"
+  },
+  "torchaudio": {
+    "version_specifier": "==2.5.1+cpu",
+    "skip": "True"
+  },
+  "torchtext": {
+    "version_specifier": "==0.18.0+cpu",
+    "skip": "True"
+  },
+  "torchtnt": {
+    "version_specifier": "==0.2.4",
+    "skip": "True"
+  },
+  "torchvision": {
+    "version_specifier": "==0.20.1+cpu",
+    "skip": "True"
+  },
+  "idna": {
+    "version_specifier": ">=3.7"
+  },
+  "tqdm": {
+    "version_specifier": ">=4.66.3"
+  },
+  "requests": {
+    "version_specifier": ">=2.32.0"
+  },
+  "setuptools": {
+    "version_specifier": ">=70.0.0"
+  },
+  "urllib3": {
+    "version_specifier": "<2"
+  },
+  "awscli": {
+    "version_specifier": "<2"
+  },
+  "numpy": {
+    "version_specifier": "<2"
+  },
+  "sagemaker": {
+    "version_specifier": ">=2,<3"
+  },
+  "sagemaker-experiments": {
+    "version_specifier": "<1"
+  },
+  "cloudpickle": {
+    "version_specifier": "==2.2.1"
+  }
+}
diff --git a/pytorch/training/docker/2.5/py3/cu124/Dockerfile.ec2.gpu.core_packages.json b/pytorch/training/docker/2.5/py3/cu124/Dockerfile.ec2.gpu.core_packages.json
new file mode 100644
index 000000000000..7b21d99c299d
--- /dev/null
+++ b/pytorch/training/docker/2.5/py3/cu124/Dockerfile.ec2.gpu.core_packages.json
@@ -0,0 +1,59 @@
+{
+  "accelerate": {
+    "version_specifier": "==1.1.1",
+    "skip": "True"
+  },
+  "fastai": {
+    "version_specifier": "==2.7.18",
+    "skip": "True"
+  },
+  "flash-attn": {
+    "version_specifier": "==2.6.3",
+    "skip": "True"
+  },
+  "transformer-engine": {
+    "version_specifier": "==1.11.0+fc03478",
+    "skip": "True"
+  },
+  "s3torchconnector": {
+    "version_specifier": "==1.2.6",
+    "skip": "True"
+  },
+  "torchaudio": {
+    "version_specifier": "==2.5.1+cu124",
+    "skip": "True"
+  },
+  "torchtext": {
+    "version_specifier": "==0.18.0+cu124",
+    "skip": "True"
+  },
+  "torchtnt": {
+    "version_specifier": "==0.2.4",
+    "skip": "True"
+  },
+  "torchvision": {
+    "version_specifier": "==0.20.1+cu124",
+    "skip": "True"
+  },
+  "idna": {
+    "version_specifier": ">=3.7"
+  },
+  "tqdm": {
+    "version_specifier": ">=4.66.3"
+  },
+  "requests": {
+    "version_specifier": ">=2.32.0"
+  },
+  "setuptools": {
+    "version_specifier": ">=70.0.0"
+  },
+  "urllib3": {
+    "version_specifier": "<2"
+  },
+  "awscli": {
+    "version_specifier": "<2"
+  },
+  "numpy": {
+    "version_specifier": "<2"
+  }
+}
diff --git a/pytorch/training/docker/2.5/py3/cu124/Dockerfile.sagemaker.gpu.core_packages.json b/pytorch/training/docker/2.5/py3/cu124/Dockerfile.sagemaker.gpu.core_packages.json
new file mode 100644
index 000000000000..5b63d98140a7
--- /dev/null
+++ b/pytorch/training/docker/2.5/py3/cu124/Dockerfile.sagemaker.gpu.core_packages.json
@@ -0,0 +1,68 @@
+{
+  "accelerate": {
+    "version_specifier": "==1.1.1",
+    "skip": "True"
+  },
+  "fastai": {
+    "version_specifier": "==2.7.18",
+    "skip": "True"
+  },
+  "flash-attn": {
+    "version_specifier": "==2.6.3",
+    "skip": "True"
+  },
+  "transformer-engine": {
+    "version_specifier": "==1.11.0+fc03478",
+    "skip": "True"
+  },
+  "s3torchconnector": {
+    "version_specifier": "==1.2.6",
+    "skip": "True"
+  },
+  "torchaudio": {
+    "version_specifier": "==2.5.1+cu124",
+    "skip": "True"
+  },
+  "torchtext": {
+    "version_specifier": "==0.18.0+cu124",
+    "skip": "True"
+  },
+  "torchtnt": {
+    "version_specifier": "==0.2.4",
+    "skip": "True"
+  },
+  "torchvision": {
+    "version_specifier": "==0.20.1+cu124",
+    "skip": "True"
+  },
+  "idna": {
+    "version_specifier": ">=3.7"
+  },
+  "tqdm": {
+    "version_specifier": ">=4.66.3"
+  },
+  "requests": {
+    "version_specifier": ">=2.32.0"
+  },
+  "setuptools": {
+    "version_specifier": ">=70.0.0"
+  },
+  "urllib3": {
+    "version_specifier": "<2"
+  },
+  "awscli": {
+    "version_specifier": "<2"
+  },
+  "numpy": {
+    "version_specifier": "<2"
+  },
+  "sagemaker": {
+    "version_specifier": ">=2,<3"
+  },
+  "sagemaker-experiments": {
+    "version_specifier": "<1"
+  },
+  "cloudpickle": {
+    "version_specifier": "==2.2.1"
+  }
+}