github.com/kubeflow/training-operator@v1.7.0/examples/pytorch/elastic/echo/Dockerfile (about)

     1  FROM python:3.8-buster
     2  WORKDIR /workspace
     3  RUN pip install torch==1.13.1 numpy
     4  # TODO Replace this with the PIP version when available
     5  ADD examples/pytorch/elastic/echo/echo.py echo.py
     6  ENV PYTHONPATH /workspace
     7  ENV ALLOW_NONE_AUTHENTICATION yes
     8  ENTRYPOINT ["python", "-m", "torch.distributed.run"]