github.com/kubeflow/training-operator@v1.7.0/examples/tensorflow/distribution_strategy/keras-API/Dockerfile (about)

     1  FROM python:3.9
     2  
     3  RUN pip install tensorflow==2.11.0 tensorflow_datasets==4.7.0
     4  
     5  COPY examples/tensorflow/distribution_strategy/keras-API/multi_worker_strategy-with-keras.py /
     6  ENTRYPOINT ["python", "/multi_worker_strategy-with-keras.py", "--saved_model_dir", "/train/saved_model/", "--checkpoint_dir", "/train/checkpoint"]