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

     1  ARG BASE_IMAGE=pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
     2  FROM $BASE_IMAGE
     3  
     4  WORKDIR /workspace
     5  
     6  # download imagenet tiny for data
     7  RUN apt-get -q update && apt-get -q install -y wget unzip
     8  RUN wget -q http://cs231n.stanford.edu/tiny-imagenet-200.zip && unzip -q tiny-imagenet-200.zip -d data && rm tiny-imagenet-200.zip
     9  
    10  COPY examples/pytorch/elastic/imagenet/ ./examples
    11  
    12  USER root
    13  ENTRYPOINT ["python", "-m", "torch.distributed.run"]
    14  CMD ["--help"]