1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-01-07 20:36:48 +01:00
immich/machine-learning/export/ann/Dockerfile
2024-07-12 16:50:46 -04:00

28 lines
753 B
Docker

FROM mambaorg/micromamba:bookworm-slim@sha256:333f7598ff2c2400fb10bfe057709c68b7daab5d847143af85abcf224a07271a as builder
ENV TRANSFORMERS_CACHE=/cache \
PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1 \
PATH="/opt/venv/bin:$PATH"
WORKDIR /export/ann
USER root
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
curl \
git
USER $MAMBA_USER
COPY --chown=$MAMBA_USER:$MAMBA_USER env.yaml ./
RUN micromamba install -y -f env.yaml
COPY --chown=$MAMBA_USER:$MAMBA_USER *.sh *.cpp ./
ENV ARMNN_PATH=/export/ann/armnn
RUN ./download-armnn.sh && \
./build-converter.sh && \
./build.sh
COPY --chown=$MAMBA_USER:$MAMBA_USER run.py ./
ENTRYPOINT ["/usr/local/bin/_entrypoint.sh"]
CMD ["python", "run.py"]