Patch tf2.18 SM image (#5299)

* patch tf training 2.18 ec2

* patch tf training 2.18 sm

* autopatch tf 2.18 sm

* autopatch tf 2.18 sm

* add docker login

* add docker login

* retest tensorflow 2.18

* build tf 2.18

* build tf2.18

* revert toml

Author: Jyothirmaikottu

tensorflow/training/docker/2.18/py3/Dockerfile.cpu

@@ -279,8 +279,9 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
 RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-experiments==0.1.45
 
+#pinning old version because of protobuf dependency with tensorflow-metadata
 RUN $PYTHON -m pip install --no-cache-dir -U \
-    sagemaker-training
+    sagemaker-training==4.8.4
 
     RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-tensorflow-training==20.4.1

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

@@ -18,7 +18,7 @@
     "version_specifier": "==20.4.1"
   },
   "sagemaker-training": {
-    "version_specifier": ">=4.8.3"
+    "version_specifier": "==4.8.4"
   },
   "sagemaker-studio-analytics-extension": {
     "version_specifier": "<1"

tensorflow/training/docker/2.18/py3/cu125/Dockerfile.gpu

@@ -383,7 +383,7 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-experiments==0.1.45
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
-    sagemaker-training
+    sagemaker-training==4.8.4
 
     RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-tensorflow-training==20.4.1

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

@@ -18,7 +18,7 @@
     "version_specifier": "==20.4.1"
   },
   "sagemaker-training": {
-    "version_specifier": ">=4.8.3"
+    "version_specifier": "==4.8.4"
   },
   "sagemaker-studio-analytics-extension": {
     "version_specifier": "<1"