fix: Mac M1 compatible Dockerfile and bump TF version (#1700)

kubeflow · Dec 19, 2022 · 52eb491 · 52eb491
1 parent ec53930
commit 52eb491
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/tensorflow/distribution_strategy/keras-API/Dockerfile b/examples/tensorflow/distribution_strategy/keras-API/Dockerfile
@@ -1,6 +1,6 @@
-FROM tensorflow/tensorflow:2.1.0-gpu-py3
+FROM python:3.9
 
-RUN pip install tensorflow_datasets==2.1.0
+RUN pip install tensorflow==2.11.0 tensorflow_datasets==4.7.0
 
 COPY multi_worker_strategy-with-keras.py /
 ENTRYPOINT ["python", "/multi_worker_strategy-with-keras.py", "--saved_model_dir", "/train/saved_model/", "--checkpoint_dir", "/train/checkpoint"]