diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_manylinux_2_28.sh b/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_manylinux_2_28.sh
index 8f39d81af2c91d..5665ea5a1ae744 100755
--- a/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_manylinux_2_28.sh
+++ b/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_manylinux_2_28.sh
@@ -91,3 +91,4 @@ python3 -m pip install --user virtualenv
 
 # Disable the cache dir to save image space, and install packages
 python3 -m pip install --no-cache-dir -r $REQUIREMENTS -U
+python3 -m pip install --no-cache-dir --no-deps tf-keras==2.17.0rc0
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_ubuntu.sh b/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_ubuntu.sh
index c2fe0d40923e3c..96077dd3d0919e 100755
--- a/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_ubuntu.sh
+++ b/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_ubuntu.sh
@@ -83,4 +83,4 @@ python3 --version
 echo "Install Requirements"
 # Disable the cache dir to save image space, and install packages
 python3 -m pip install --no-cache-dir -r $REQUIREMENTS -U
-python3 -m pip install --no-cache-dir --no-deps tf-keras-nightly
+python3 -m pip install --no-cache-dir --no-deps tf-keras==2.17.0rc0
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/setup.python.sh b/tensorflow/tools/tf_sig_build_dockerfiles/setup.python.sh
index bef0c6f32981f0..36546a511fcdff 100755
--- a/tensorflow/tools/tf_sig_build_dockerfiles/setup.python.sh
+++ b/tensorflow/tools/tf_sig_build_dockerfiles/setup.python.sh
@@ -92,10 +92,10 @@ python3 -m pip install -U setuptools
 
 if [[ $3 ]]; then
     echo "Runtime mode"
-    python3 -m pip install --no-cache-dir --no-deps tf-keras-nightly
+    python3 -m pip install --no-cache-dir --no-deps tf-keras==2.17.0rc0
 else
     echo "Install Requirements"
     # Disable the cache dir to save image space, and install packages
     python3 -m pip install --no-cache-dir -r $REQUIREMENTS -U
-    python3 -m pip install --no-cache-dir --no-deps tf-keras-nightly
+    python3 -m pip install --no-cache-dir --no-deps tf-keras==2.17.0rc0
 fi