TF2.19 SageMaker Training Autopatch

tensorflow/training/buildspec-2-19-sm.yml

@@ -5,7 +5,7 @@ framework: &FRAMEWORK tensorflow
 version: &VERSION 2.19.0
 short_version: &SHORT_VERSION "2.19"
 arch_type: x86
-#autopatch_build: "True"
+autopatch_build: "True"
 
 repository_info:
   training_repository: &TRAINING_REPOSITORY

tensorflow/training/docker/2.19/py3/Dockerfile.sagemaker.cpu.core_packages.json

@@ -0,0 +1,37 @@
+{
+  "awscli": {
+    "version_specifier": "<2"
+  },
+  "opencv-python": {
+    "version_specifier": "==4.12.0.88",
+    "skip": "True"
+  },
+  "cython": {
+    "version_specifier": "<3.0"
+  },
+  "pyyaml": {
+    "version_specifier": ">=6.0,<6.1"
+  },
+  "sparkmagic": {
+    "version_specifier": "<1"
+  },
+  "tensorflow-io": {
+    "version_specifier": "==0.37.*"
+  },
+  "tensorflow-text": {
+    "version_specifier": "==2.19.0",
+    "skip": "True"
+  },
+  "tf-keras": {
+    "version_specifier": "==2.19.0",
+    "skip": "True"
+  },
+  "tensorflow-hub": {
+    "version_specifier": "==0.16.1",
+    "skip": "True"
+  },
+  "tf-models-official": {
+    "version_specifier": "==2.19.1",
+    "skip": "True"
+  }
+}

tensorflow/training/docker/2.19/py3/cu125/Dockerfile.sagemaker.gpu.core_packages.json

@@ -0,0 +1,39 @@
+{
+  "awscli": {
+    "version_specifier": "<2"
+  },
+  "opencv-python": {
+    "version_specifier": "==4.11.0.86"
+  },
+  "cython": {
+    "version_specifier": "<3.0"
+  },
+  "pyyaml": {
+    "version_specifier": ">=6.0,<6.1"
+  },
+  "sagemaker-experiments": {
+    "version_specifier": "==0.*"
+  },
+  "sparkmagic": {
+    "version_specifier": "<1"
+  },
+  "tensorflow-hub": {
+    "version_specifier": ">=0.16.1",
+    "skip": "True"
+  },
+  "tensorflow-io": {
+    "version_specifier": "==0.37.*"
+  },
+  "tf-models-official": {
+    "version_specifier": "==2.19.1",
+    "skip": "True"
+  },
+  "tensorflow-text": {
+    "version_specifier": "==2.19.0",
+    "skip": "True"
+  },
+  "tf-keras": {
+    "version_specifier": "==2.19.0",
+    "skip": "True"
+  }
+}