rebuild gpu and cpu

Author: Jyothirmaikottu

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

@@ -287,17 +287,17 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
     plotly \
     seaborn \
     shap
-
+    
 RUN $PYTHON -m pip install --no-cache-dir -U \
-    sagemaker==2.254.1 \
-    sagemaker-experiments==0.1.45 \
+    sagemaker \
+    sagemaker-experiments \
     sagemaker-tensorflow-training==20.4.1 \
-    sagemaker-training==5.1.1 \
+    sagemaker-training \
     y-py==0.6.2 \
     sagemaker-studio-analytics-extension==0.1.7 \
     sparkmagic==0.22.0 \
     sagemaker-studio-sparkmagic-lib==0.1.4 \
-    smclarify==0.5
+    smclarify
 
 # Remove python kernel installed by sparkmagic
 RUN /usr/local/bin/jupyter-kernelspec remove -f python3

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

@@ -362,13 +362,13 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker \
-    sagemaker-experiments==0.* \
-    sagemaker-tensorflow-training \
+    sagemaker-experiments \
+    sagemaker-tensorflow-training==20.4.1 \
     sagemaker-training \
-    y-py \
-    sagemaker-studio-analytics-extension \
-    "sparkmagic<1" \
-    sagemaker-studio-sparkmagic-lib \
+    y-py==0.6.2 \
+    sagemaker-studio-analytics-extension==0.1.7 \
+    sparkmagic==0.22.0 \
+    sagemaker-studio-sparkmagic-lib==0.1.4 \
     smclarify
 
 # install boost