mirror of
https://github.com/immich-app/immich.git
synced 2024-11-27 09:21:05 +02:00
b53bd8c525
* fix(deps): update machine-learning * update openvino options, cuda * update openvino build * fix indentation * update minimum nvidia driver --------- Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com> Co-authored-by: mertalev <101130780+mertalev@users.noreply.github.com>
103 lines
3.2 KiB
Docker
103 lines
3.2 KiB
Docker
ARG DEVICE=cpu
|
|
|
|
FROM python:3.11-bookworm@sha256:7bec1574675e7fd9e3a540a03cd7d6811c59ca261bd300cd665369d8f435298a as builder-cpu
|
|
|
|
FROM builder-cpu as builder-openvino
|
|
|
|
FROM builder-cpu as builder-cuda
|
|
|
|
FROM builder-cpu as builder-armnn
|
|
|
|
ENV ARMNN_PATH=/opt/armnn
|
|
COPY ann /opt/ann
|
|
RUN mkdir /opt/armnn && \
|
|
curl -SL "https://github.com/ARM-software/armnn/releases/download/v24.05/ArmNN-linux-aarch64.tar.gz" | tar -zx -C /opt/armnn && \
|
|
cd /opt/ann && \
|
|
sh build.sh
|
|
|
|
FROM builder-${DEVICE} as builder
|
|
|
|
ARG DEVICE
|
|
ENV PYTHONDONTWRITEBYTECODE=1 \
|
|
PYTHONUNBUFFERED=1 \
|
|
PIP_NO_CACHE_DIR=true \
|
|
VIRTUAL_ENV="/opt/venv" \
|
|
PATH="/opt/venv/bin:${PATH}"
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends g++
|
|
|
|
RUN pip install --upgrade pip && pip install poetry
|
|
RUN poetry config installer.max-workers 10 && \
|
|
poetry config virtualenvs.create false
|
|
RUN python3 -m venv /opt/venv
|
|
|
|
COPY poetry.lock pyproject.toml ./
|
|
RUN poetry install --sync --no-interaction --no-ansi --no-root --with ${DEVICE} --without dev
|
|
|
|
FROM python:3.11-slim-bookworm@sha256:17ec9dc2367aa748559d0212f34665ec4df801129de32db705ea34654b5bc77a as prod-cpu
|
|
|
|
FROM prod-cpu as prod-openvino
|
|
|
|
COPY scripts/configure-apt.sh ./
|
|
RUN ./configure-apt.sh && \
|
|
apt-get update && \
|
|
apt-get install -t unstable --no-install-recommends -yqq intel-opencl-icd && \
|
|
rm configure-apt.sh
|
|
|
|
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04@sha256:fa44193567d1908f7ca1f3abf8623ce9c63bc8cba7bcfdb32702eb04d326f7a8 as prod-cuda
|
|
|
|
COPY --from=builder-cuda /usr/local/bin/python3 /usr/local/bin/python3
|
|
COPY --from=builder-cuda /usr/local/lib/python3.11 /usr/local/lib/python3.11
|
|
COPY --from=builder-cuda /usr/local/lib/libpython3.11.so /usr/local/lib/libpython3.11.so
|
|
|
|
FROM prod-cpu as prod-armnn
|
|
|
|
ENV LD_LIBRARY_PATH=/opt/armnn
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends ocl-icd-libopencl1 mesa-opencl-icd libgomp1 && \
|
|
rm -rf /var/lib/apt/lists/* && \
|
|
mkdir --parents /etc/OpenCL/vendors && \
|
|
echo "/usr/lib/libmali.so" > /etc/OpenCL/vendors/mali.icd && \
|
|
mkdir /opt/armnn
|
|
|
|
COPY --from=builder-armnn \
|
|
/opt/armnn/libarmnn.so.?? \
|
|
/opt/armnn/libarmnnOnnxParser.so.?? \
|
|
/opt/armnn/libarmnnDeserializer.so.?? \
|
|
/opt/armnn/libarmnnTfLiteParser.so.?? \
|
|
/opt/armnn/libprotobuf.so.?.??.?.? \
|
|
/opt/ann/libann.s[o] \
|
|
/opt/ann/build.sh \
|
|
/opt/armnn/
|
|
|
|
FROM prod-${DEVICE} as prod
|
|
ARG DEVICE
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y --no-install-recommends tini $(if ! [ "$DEVICE" = "openvino" ]; then echo "libmimalloc2.0"; fi) && \
|
|
apt-get autoremove -yqq && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /usr/src/app
|
|
ENV TRANSFORMERS_CACHE=/cache \
|
|
PYTHONDONTWRITEBYTECODE=1 \
|
|
PYTHONUNBUFFERED=1 \
|
|
PATH="/opt/venv/bin:$PATH" \
|
|
PYTHONPATH=/usr/src \
|
|
DEVICE=${DEVICE}
|
|
|
|
# prevent core dumps
|
|
RUN echo "hard core 0" >> /etc/security/limits.conf && \
|
|
echo "fs.suid_dumpable 0" >> /etc/sysctl.conf && \
|
|
echo 'ulimit -S -c 0 > /dev/null 2>&1' >> /etc/profile
|
|
|
|
COPY --from=builder /opt/venv /opt/venv
|
|
COPY ann/ann.py /usr/src/ann/ann.py
|
|
COPY start.sh log_conf.json ./
|
|
COPY app .
|
|
ENTRYPOINT ["tini", "--"]
|
|
CMD ["./start.sh"]
|
|
|
|
HEALTHCHECK CMD python3 healthcheck.py
|