Fix tf_keras (#2710)

Author: jayfurmanek

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.manylinux_2_28

@@ -63,9 +63,6 @@ RUN mkdir /tmp/llvm-project && wget -qO - https://github.com/llvm/llvm-project/a
 COPY devel.usertools /usertools
 COPY devel.bashrc /root/.bashrc
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 # Setup ENV variables for tensorflow pip build
 ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt.ub20

@@ -40,9 +40,6 @@ ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1
 ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 RUN pip install --no-cache-dir /${TF_PKGS_DIR}/${TENSORFLOW_PACKAGE}
 RUN echo 'ALL ALL=NOPASSWD:ALL' | tee /etc/sudoers.d/sudo-nopasswd
 

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt.ub22

@@ -36,9 +36,6 @@ ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1
 ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 RUN pip install --no-cache-dir /${TF_PKGS_DIR}/${TENSORFLOW_PACKAGE}
 RUN echo 'ALL ALL=NOPASSWD:ALL' | tee /etc/sudoers.d/sudo-nopasswd
 

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt.ub24

@@ -36,9 +36,6 @@ ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1
 ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 RUN pip install --no-cache-dir /${TF_PKGS_DIR}/${TENSORFLOW_PACKAGE}
 RUN echo 'ALL ALL=NOPASSWD:ALL' | tee /etc/sudoers.d/sudo-nopasswd
 

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub20

@@ -54,9 +54,6 @@ RUN if [ -n "${DWLD_TF_SRC_CMD}" ]; then eval "${DWLD_TF_SRC_CMD}"; fi
 COPY devel.usertools /usertools
 COPY devel.bashrc /root/.bashrc
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 # Setup ENV variables for tensorflow pip build
 ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub22

@@ -54,9 +54,6 @@ RUN if [ -n "${DWLD_TF_SRC_CMD}" ]; then eval "${DWLD_TF_SRC_CMD}"; fi
 COPY devel.usertools /usertools
 COPY devel.bashrc /root/.bashrc
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 # Setup ENV variables for tensorflow pip build
 ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1

tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub24

@@ -54,9 +54,6 @@ RUN if [ -n "${DWLD_TF_SRC_CMD}" ]; then eval "${DWLD_TF_SRC_CMD}"; fi
 COPY devel.usertools /usertools
 COPY devel.bashrc /root/.bashrc
 
-# Install tf_keras
-RUN pip install --no-deps tf_keras==2.17.0
-
 # Setup ENV variables for tensorflow pip build
 ENV TF_NEED_ROCM=1
 ENV TF_ROCM_GCC=1

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.0

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.0

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.0
 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.0
 fi