immich/machine-learning/Dockerfile

FROM python:3.10 as builder

ENV PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=true

RUN python -m venv /opt/venv
RUN /opt/venv/bin/pip install --pre torch  -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
RUN /opt/venv/bin/pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow gunicorn
RUN /opt/venv/bin/pip install --no-deps sentence-transformers

FROM python:3.10-slim

COPY --from=builder /opt/venv /opt/venv

ENV TRANSFORMERS_CACHE=/cache \
    PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PATH="/opt/venv/bin:$PATH"

WORKDIR /usr/src/app

COPY . .

CMD ["gunicorn", "src.main:server"]
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-04 00:45:20 +02:00			`FROM python:3.10 as builder`
fix(server): Flask not found (#1830) * fix(server): Flask not found * trial 1 2023-02-22 18:53:59 +02:00
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-04 00:45:20 +02:00			`ENV PYTHONDONTWRITEBYTECODE=1 \`
feat(server): Machine learning's image optimisations (#1908) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable 2023-03-01 17:37:12 +02:00			`PYTHONUNBUFFERED=1 \`
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 19:48:35 +02:00			`PIP_NO_CACHE_DIR=true`
feat(server): Machine learning's image optimisations (#1908) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable 2023-03-01 17:37:12 +02:00
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 19:48:35 +02:00			`RUN python -m venv /opt/venv`
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-04 00:45:20 +02:00			`RUN /opt/venv/bin/pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html`
			`RUN /opt/venv/bin/pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow gunicorn`
			`RUN /opt/venv/bin/pip install --no-deps sentence-transformers`

			`FROM python:3.10-slim`

			`COPY --from=builder /opt/venv /opt/venv`
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 19:48:35 +02:00
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-04 00:45:20 +02:00			`ENV TRANSFORMERS_CACHE=/cache \`
			`PYTHONDONTWRITEBYTECODE=1 \`
			`PYTHONUNBUFFERED=1 \`
			`PATH="/opt/venv/bin:$PATH"`

			`WORKDIR /usr/src/app`
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 19:48:35 +02:00
Added machine learning microservice and object detection (#76) 2022-03-27 21:58:54 +02:00			`COPY . .`
fix(machine-learning): Add the command to execute at startup (#1843) * fix(machine-learning): Add the command to execute at startup Previously it wasn't set in the Docker container but it should be. * fix(docker): remove machine-learning command arg * fix(docker): machine-learning CMD argument 2023-02-23 17:54:04 +02:00
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-04 00:45:20 +02:00			`CMD ["gunicorn", "src.main:server"]`