decrease version sm tf training

Author: Jyothirmaikottu

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

@@ -264,14 +264,14 @@ RUN ${PIP} install --no-cache-dir -U \
     "tensorflow-datasets==4.9.7"
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
-    numba==0.60.0 \
+    numba==0.61.0 \
     bokeh \
     imageio \
-    opencv-python==4.8.0.76 \
+    opencv-python \
     plotly \
     seaborn \
     shap \
-    numpy==1.23.5
+    numpy
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
     "sagemaker<3"
@@ -280,7 +280,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==4.8.3
+    sagemaker-training
 
     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

@@ -6,7 +6,7 @@
     "version_specifier": "<3.0"
   },
   "protobuf": {
-    "version_specifier": "==3.20.3"
+    "version_specifier": ">=3.20.3"
   },
   "pyyaml": {
     "version_specifier": ">=6.0,<6.1"
@@ -18,7 +18,7 @@
     "version_specifier": "==20.4.1"
   },
   "sagemaker-training": {
-    "version_specifier": "==4.8.3"
+    "version_specifier": ">=4.8.3"
   },
   "sagemaker-studio-analytics-extension": {
     "version_specifier": "<1"

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

@@ -367,14 +367,14 @@ RUN ${PIP} install --no-cache-dir -U \
     "tensorflow-datasets==4.9.7"
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
-    numba==0.60.0 \
+    numba==0.61.0 \
     bokeh \
     imageio \
-    opencv-python==4.8.0.76 \
+    opencv-python \
     plotly \
     seaborn \
     shap \
-    numpy==1.23.5
+    numpy
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
     "sagemaker<3"
@@ -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==4.8.3
+    sagemaker-training
 
     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

@@ -6,7 +6,7 @@
     "version_specifier": "<3.0"
   },
   "protobuf": {
-    "version_specifier": "==3.20.3"
+    "version_specifier": ">=3.20.3"
   },
   "pyyaml": {
     "version_specifier": ">=6.0,<6.1"
@@ -18,7 +18,7 @@
     "version_specifier": "==20.4.1"
   },
   "sagemaker-training": {
-    "version_specifier": "==4.8.3"
+    "version_specifier": ">=4.8.3"
   },
   "sagemaker-studio-analytics-extension": {
     "version_specifier": "<1"