https://github.com/kubeflow/katib
Raw File
Tip revision: f99736db4c79009e8988b498090116b5b09230bb authored by c-bata on 09 April 2020, 14:11:54 UTC
Avoid duplicated import problems
Tip revision: f99736d
Dockerfile
FROM pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime

WORKDIR /var
ADD mnist.py /var

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