github/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-06-07 16:30:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ml): healthcheck (#17274)

Browse Source
This commit is contained in:
Mert 2025-03-31 19:23:40 -04:00 committed by GitHub
parent 7729fe80fa
commit 9e7744a9ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
machine-learning/Dockerfile
View File

@ -67,7 +67,7 @@ FROM python:3.11-slim-bookworm@sha256:7029b00486ac40bed03e36775b864d3f3d39dcbdf1
ENV LD_PRELOAD=/usr/lib/libmimalloc.so.2 ENV LD_PRELOAD=/usr/lib/libmimalloc.so.2
FROM prod-cpu AS prod-openvino FROM python:3.11-slim-bookworm@sha256:7029b00486ac40bed03e36775b864d3f3d39dcbdf19cd45e6a52d541e6c178f0 AS prod-openvino
RUN apt-get update && \ RUN apt-get update && \
apt-get install --no-install-recommends -yqq ocl-icd-libopencl1 wget && \ apt-get install --no-install-recommends -yqq ocl-icd-libopencl1 wget && \
@ -82,6 +82,8 @@ RUN apt-get update && \
FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04@sha256:94c1577b2cd9dd6c0312dc04dff9cb2fdce2b268018abc3d7c2dbcacf1155000 AS prod-cuda FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04@sha256:94c1577b2cd9dd6c0312dc04dff9cb2fdce2b268018abc3d7c2dbcacf1155000 AS prod-cuda
ENV LD_PRELOAD=/usr/lib/libmimalloc.so.2
RUN apt-get update && \ RUN apt-get update && \
apt-get install --no-install-recommends -yqq libcudnn9-cuda-12 && \ apt-get install --no-install-recommends -yqq libcudnn9-cuda-12 && \
apt-get clean && \ apt-get clean && \
@ -140,7 +142,6 @@ ENV TRANSFORMERS_CACHE=/cache \
PYTHONPATH=/usr/src \ PYTHONPATH=/usr/src \
DEVICE=${DEVICE} \ DEVICE=${DEVICE} \
VIRTUAL_ENV=/opt/venv \ VIRTUAL_ENV=/opt/venv \
LD_BIND_NOW=1 \
MACHINE_LEARNING_CACHE_FOLDER=/cache MACHINE_LEARNING_CACHE_FOLDER=/cache
# prevent core dumps # prevent core dumps
@ -149,6 +150,7 @@ RUN echo "hard core 0" >> /etc/security/limits.conf && \
echo 'ulimit -S -c 0 > /dev/null 2>&1' >> /etc/profile echo 'ulimit -S -c 0 > /dev/null 2>&1' >> /etc/profile
COPY --from=builder /opt/venv /opt/venv COPY --from=builder /opt/venv /opt/venv
COPY scripts/healthcheck.py .
COPY immich_ml immich_ml COPY immich_ml immich_ml
ARG BUILD_ID ARG BUILD_ID