https://github.com/kubeflow/katib
Raw File
Tip revision: 1d7e7d85da55b72f1f958fe5b4ad32b416339b30 authored by avelichk on 24 July 2020, 23:36:42 UTC
Update git ignore
Tip revision: 1d7e7d8
Dockerfile
FROM pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime

RUN mkdir -p /katib /opt/mnist/src

WORKDIR /opt/mnist/src
ADD mnist.py /opt/mnist/src

RUN chgrp -R 0 /opt/mnist \
  && chmod -R g+rwX /opt/mnist \
  && chgrp -R 0 /katib \
  && chmod -R g+rwX /katib

ENTRYPOINT ["python", "/opt/mnist/src/mnist.py"]
back to top