wise/Dockerfile at wise2 · ox-vgg/wise · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# Based on
# https://micromamba-docker.readthedocs.io/en/latest/advanced_usage.html
# https://uwekorn.com/2021/03/01/deploying-conda-environments-in-docker-how-to-do-it-right.html
ARG MAMBA_IMAGE=mambaorg/micromamba:1.5.10
ARG NODE_IMAGE=node:22-slim
ARG PYTHON_IMAGE=python:3.10-slim

# For holding huggingface cache, and accessible by any user in the system who runs the container
ARG CACHE_DIR=/tmp/tmp

# Can be wise / wise-cpu
ARG APP=wise

FROM ${MAMBA_IMAGE} AS wise-env
ARG APP
ARG CACHE_DIR
ENV APP=${APP} \
    CACHE_DIR=${CACHE_DIR} \
    HF_HOME=${CACHE_DIR}/hfcache  \
    PYTHONPYCACHEPREFIX=${CACHE_DIR}/pycache


USER root
RUN mkdir -p ${HF_HOME} ${PYTHONPYCACHEPREFIX} && chmod 3777 -R /tmp
RUN apt-get update && apt-get install -y --no-install-recommends \
      build-essential && \
    rm -rf /var/lib/apt /var/lib/dpkg /var/lib/cache /var/lib/log

USER ${MAMBA_USER}
WORKDIR /tmp

COPY --chown=${MAMBA_USER}:${MAMBA_USER} "docker/wise.yml" ./
RUN --mount=type=cache,target=/opt/conda/pkgs \
    --mount=type=cache,target=/home/${MAMBA_USER}/.cache,uid=${MAMBA_USER_ID},gid=${MAMBA_USER_GID} \
    export PIP_EXTRA_INDEX_URL='https://download.pytorch.org/whl/cpu' && \
    export CUDNN_PACKAGE='' && \
    export ONNXRUNTIME_PACKAGE='onnxruntime' && \
    if [[ ${APP} == 'wise' ]]; then \
        export PIP_EXTRA_INDEX_URL='https://download.pytorch.org/whl/cu124' && \
        export CUDNN_PACKAGE='cudnn' && \
        export ONNXRUNTIME_PACKAGE='onnxruntime-gpu'; \
    fi && \
    echo "Using Pip index url: ${PIP_EXTRA_INDEX_URL} CUDNN: ${CUDNN_PACAKGE:-none} ONNX: ${ONNXRUNTIME_PACKAGE} " && \
    micromamba create --always-copy --yes -n wise-env -f "wise.yml" "${CUDNN_PACKAGE}" && \
    ${MAMBA_ROOT_PREFIX}/envs/wise-env/bin/python3 -m pip install "${ONNXRUNTIME_PACKAGE}"

FROM ${NODE_IMAGE} AS wise-frontend

RUN mkdir -p /wise/frontend/node_modules && \
    chown -R node:node /wise/frontend
COPY frontend/package*.json /wise/frontend/
WORKDIR /wise/frontend
USER node
RUN npm ci
COPY --chown=node:node frontend .
RUN npm run build

FROM gcr.io/distroless/base-debian12 AS wise
ARG APP
ARG CACHE_DIR
ENV APP=${APP} \
    PYTHONUNBUFFERED=1 \
    PYTHONPYCACHEPREFIX=/tmp/pycache \
    HF_HOME=/tmp/hfcache \
    PATH="/env/bin/:${PATH}"

COPY --from=wise-env --chown=nonroot:nonroot \
    /opt/conda/envs/wise-env /env

WORKDIR /wise
COPY --from=wise-env --chmod=7777 --chown=nonroot:nonroot  \
    ${CACHE_DIR}/ /tmp

COPY --chown=nonroot:nonroot . .

COPY --from=wise-frontend --chown=nonroot:nonroot --chmod=3775 \
    /wise/frontend/dist/ /wise/frontend/dist/

# Hack to allow the rewrite of index.html when we serve the project
# The user who runs the container is possibly not nonroot, so we need a+rw equivalent
COPY --from=wise-frontend --chown=nonroot:nonroot --chmod=0666 \
    /wise/frontend/dist/index.html /wise/frontend/dist/index.html

# For libmagic to find the database in a non standard location
ENV MAGIC='/env/share/misc/magic' \
    LD_LIBRARY_PATH="/env/lib/:${LD_LIBRARY_PATH}"

# You can modify the CMD statement as needed....
ENTRYPOINT ["python3"]
VOLUME [ "/tmp" ]
CMD ["extract-features.py", "--help"]