pin tf-keras for 2.16

ROCm · Oct 11, 2024 · b61eb7f · b61eb7f
1 parent 6b10bbd
commit b61eb7f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
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
@@ -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.16.0
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
@@ -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.16.0