From b8856bc756619cb1b88772c15cdabffaa061f281 Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Thu, 15 Dec 2022 13:06:20 -0500 Subject: [PATCH] Update Dockerfile --- .../tensorflow/distribution_strategy/keras-API/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/tensorflow/distribution_strategy/keras-API/Dockerfile b/examples/tensorflow/distribution_strategy/keras-API/Dockerfile index 910c39d3da..3dc714dd54 100644 --- 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"]