mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-10-30 14:46:14 +00:00
55550f8df5
* Out-of-the-box support TrainJob Signed-off-by: Ram Lau <[email protected]> * Example for Pytorch Distributed Signed-off-by: Ram Lau <[email protected]> * Update examples/v1beta1/kubeflow-training-operator/trainjob-pytorch.yaml Co-authored-by: Andrey Velichkevich <[email protected]> Signed-off-by: Ram Lau <[email protected]> * Create folder for Trainer as suggested Signed-off-by: Ram Lau <[email protected]> * Movethe exmaple of trainjob to the new folder Signed-off-by: Ram Lau <[email protected]> * Ref the primaryContainerName to that of ClusterTrainingRuntime Signed-off-by: Ram Lau <[email protected]> * tenzen-y steps down from Katib approver role (#2561) Signed-off-by: Yuki Iwai <[email protected]> Signed-off-by: Ram Lau <[email protected]> * Set Default value for TrainJob Success, Failure Condition and PrimaryPodLabels in the trial Template Signed-off-by: Ram Lau <[email protected]> * Enchance Handling for default value of Success, Fail Cond & Pod Label Signed-off-by: Ram Lau <[email protected]> * Bug fix for default value condition Signed-off-by: Ram Lau <[email protected]> * code format by hack/update-gofmt.sh Signed-off-by: Ram Lau <[email protected]> * add TrainJob trial Resources to cert manager config Signed-off-by: Ram Lau <[email protected]> * add trainjob to controller rbac Signed-off-by: Ram Lau <[email protected]> * Grant JobSet permission to Katib controller Signed-off-by: Andrey Velichkevich <[email protected]> * Remove create/delete RBAC for TrainJob Signed-off-by: Andrey Velichkevich <[email protected]> * Fix docker build with libpcre2 Signed-off-by: Andrey Velichkevich <[email protected]> --------- Signed-off-by: Ram Lau <[email protected]> Signed-off-by: Yuki Iwai <[email protected]> Signed-off-by: Andrey Velichkevich <[email protected]> Co-authored-by: Andrey Velichkevich <[email protected]> Co-authored-by: Yuki Iwai <[email protected]>
23 lines
560 B
Docker
23 lines
560 B
Docker
FROM python:3.11-slim
|
|
|
|
ARG TARGETARCH
|
|
ENV TARGET_DIR /opt/enas-cnn-cifar10
|
|
ENV PYTHONPATH ${TARGET_DIR}
|
|
|
|
ADD examples/v1beta1/trial-images/enas-cnn-cifar10 ${TARGET_DIR}
|
|
|
|
WORKDIR ${TARGET_DIR}
|
|
|
|
RUN if [ "${TARGETARCH}" = "arm64" ]; then \
|
|
apt-get -y update && \
|
|
apt-get -y install gfortran libpcre2-dev && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*; \
|
|
fi
|
|
|
|
RUN pip install --prefer-binary --no-cache-dir -r requirements.txt
|
|
RUN chgrp -R 0 ${TARGET_DIR} \
|
|
&& chmod -R g+rwX ${TARGET_DIR}
|
|
|
|
ENTRYPOINT ["python3", "-u", "RunTrial.py"]
|