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 fastapi Pillow uvicorn[standard]
RUN /opt/venv/bin/pip install --no-deps sentence-transformers


FROM python:3.10-slim

ENV NODE_ENV=production

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 . .
ENV PYTHONPATH=`pwd`
CMD ["python", "main.py"]
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 \`
chore(ml): move to fastAPI (#2336) 2023-04-26 12:39:24 +02:00			`PYTHONUNBUFFERED=1 \`
			`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`
chore(ml): move to fastAPI (#2336) 2023-04-26 12:39:24 +02:00			`RUN /opt/venv/bin/pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece fastapi Pillow uvicorn[standard]`
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 --no-deps sentence-transformers`

chore(ml): move to fastAPI (#2336) 2023-04-26 12:39:24 +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			`FROM python:3.10-slim`

chore(docker): Default NODE_ENV to production for server image and update alpine version (#2157) * default NODE_ENV to production for server image * update node image to use 3.17 alpine in server * update web docker image to use alpine 3.17 * remove NODE_ENV from production docker-compose * NODE_ENV is also needed default in machine-learning 2023-04-03 22:05:29 +02:00			`ENV NODE_ENV=production`

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			`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 \`
chore(ml): move to fastAPI (#2336) 2023-04-26 12:39:24 +02:00			`PYTHONDONTWRITEBYTECODE=1 \`
			`PYTHONUNBUFFERED=1 \`
			`PATH="/opt/venv/bin:$PATH"`
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
			`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 . .`
chore(ml): move to fastAPI (#2336) 2023-04-26 12:39:24 +02:00			ENV PYTHONPATH=`pwd`
			`CMD ["python", "main.py"]`