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

     1  FROM pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime
     2  
     3  RUN pip install tensorboardX==1.6.0
     4  RUN mkdir -p /opt/mnist
     5  
     6  WORKDIR /opt/mnist/src
     7  ADD mnist.py /opt/mnist/src/mnist.py
     8  
     9  RUN  chgrp -R 0 /opt/mnist \
    10    && chmod -R g+rwX /opt/mnist
    11  
    12  ENTRYPOINT ["python", "/opt/mnist/src/mnist.py"]