add a specific version of tensorflow_datasets (#1305)

re-push with signed email
kubeflow · Jul 20, 2021 · 47a74b7 · 47a74b7
1 parent e306113
commit 47a74b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/v1/distribution_strategy/keras-API/Dockerfile b/examples/v1/distribution_strategy/keras-API/Dockerfile
@@ -1,6 +1,6 @@
 FROM tensorflow/tensorflow:2.1.0-gpu-py3
 
-RUN pip install tensorflow_datasets
+RUN pip install tensorflow_datasets==2.1.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"]