Update tf job examples to tf v2

.github/workflows/publish-example-images.yaml

@@ -24,12 +24,9 @@ jobs:
           - component-name: tf-dist-mnist-test
             platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/dist-mnist/Dockerfile
-          - component-name: tf-distributed-worker
-            platforms: linux/amd64,linux/arm64
-            dockerfile: examples/tensorflow/distribution_strategy/estimator-API/Dockerfile
           - component-name: tf-multi-worker-strategy
             platforms: linux/amd64,linux/arm64
-            dockerfile: examples/tensorflow/distribution_strategy/keras-API/Dockerfile
+            dockerfile: examples/tensorflow/distribution_strategy/Dockerfile
           - component-name: tf-mnist-with-summaries
             platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/mnist_with_summaries/Dockerfile

examples/tensorflow/dist-mnist/Dockerfile

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM tensorflow/tensorflow:1.5.0
+FROM tensorflow/tensorflow:2.17.0
+
+# Using keras-2.17 because of bug on keras-3.4.1
+# which used by default by TF-2.17 (https://github.com/tensorflow/tensorflow/issues/72388)
+ENV TF_USE_LEGACY_KERAS 1
+RUN pip install tf_keras
 
 ADD examples/tensorflow/dist-mnist/ /var/tf_dist_mnist
+ADD examples/tensorflow/mnist_utils.py /var/tf_dist_mnist
 ENTRYPOINT ["python", "/var/tf_dist_mnist/dist_mnist.py"]

examples/tensorflow/dist-mnist/README.md

@@ -10,14 +10,9 @@ To build this image on x86_64:
 ```shell
 docker build -f Dockerfile -t kubeflow/tf-dist-mnist-test:1.0 ./
 ```
-To build this image on ppc64le:
-```shell
-docker build -f Dockerfile.ppc64le -t kubeflow123/tf-dist-mnist-test:1.0 ./
-```
 
 **Create TFJob YAML**
 
 ```
 kubectl create -f ./tf_job_mnist.yaml
 ```
-  * If on ppc64le, please update tf_job_mnist.yaml to use the image of ppc64le firstly.