mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-09-24 00:31:27 +00:00
Andrey VelichkevichandGitHub
5748c7e855
* [SDK] Fix empty list for env variables Signed-off-by: Andrey Velichkevich <[email protected]> * Fix numpy version in tests Signed-off-by: Andrey Velichkevich <[email protected]> --------- Signed-off-by: Andrey Velichkevich <[email protected]>
21 lines
623 B
Docker
21 lines
623 B
Docker
FROM python:3.11-slim
|
|
|
|
ADD examples/v1beta1/trial-images/pytorch-mnist /opt/pytorch-mnist
|
|
|
|
WORKDIR /opt/pytorch-mnist
|
|
|
|
# Add folder for the logs.
|
|
RUN mkdir /katib
|
|
|
|
# TODO (andreyvelich): This is required since torchvision==0.17.1 is incompatible with numpy 2.0
|
|
RUN pip install numpy==1.26.0
|
|
RUN pip install --prefer-binary --no-cache-dir torch==2.2.1 torchvision==0.17.1
|
|
RUN pip install --prefer-binary --no-cache-dir -r requirements.txt
|
|
|
|
RUN chgrp -R 0 /opt/pytorch-mnist \
|
|
&& chmod -R g+rwX /opt/pytorch-mnist \
|
|
&& chgrp -R 0 /katib \
|
|
&& chmod -R g+rwX /katib
|
|
|
|
ENTRYPOINT ["python3", "/opt/pytorch-mnist/mnist.py"]
|