mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-09-24 00:31:27 +00:00
Yuki IwaiandGitHub
105aea3d9e
* Upgrade Python version to 3.11 Signed-off-by: Yuki Iwai <[email protected]> * Upgrade the numpy version to 1.25.2 Signed-off-by: Yuki Iwai <[email protected]> * Increase resource requests for the ENAS suggestion service Signed-off-by: Yuki Iwai <[email protected]> * Update pytest CI Signed-off-by: Yuki Iwai <[email protected]> * Prepare dedicated pytest for skopt Signed-off-by: Yuki Iwai <[email protected]> --------- Signed-off-by: Yuki Iwai <[email protected]>
25 lines
740 B
Docker
25 lines
740 B
Docker
FROM python:3.11-slim
|
|
|
|
ARG TARGETARCH
|
|
ENV TARGET_DIR /opt/katib
|
|
ENV SUGGESTION_DIR cmd/suggestion/hyperopt/v1beta1
|
|
ENV PYTHONPATH ${TARGET_DIR}:${TARGET_DIR}/pkg/apis/manager/v1beta1/python:${TARGET_DIR}/pkg/apis/manager/health/python
|
|
|
|
RUN if [ "${TARGETARCH}" = "ppc64le" ]; then \
|
|
apt-get -y update && \
|
|
apt-get -y install gfortran libopenblas-dev liblapack-dev && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*; \
|
|
fi
|
|
|
|
ADD ./pkg/ ${TARGET_DIR}/pkg/
|
|
ADD ./${SUGGESTION_DIR}/ ${TARGET_DIR}/${SUGGESTION_DIR}/
|
|
|
|
WORKDIR ${TARGET_DIR}/${SUGGESTION_DIR}
|
|
|
|
RUN pip install --prefer-binary --no-cache-dir -r requirements.txt
|
|
RUN chgrp -R 0 ${TARGET_DIR} \
|
|
&& chmod -R g+rwX ${TARGET_DIR}
|
|
|
|
ENTRYPOINT ["python", "main.py"]
|