immich/machine_learning/Dockerfile

25 lines
478 B
Docker
Raw Normal View History

## GPU Build
2022-02-19 21:57:11 -07:00
# FROM tensorflow/tensorflow:latest-gpu as gpu
2022-02-19 21:57:11 -07:00
# WORKDIR /code
2022-02-19 21:57:11 -07:00
# COPY ./requirements.txt /code/requirements.txt
2022-02-19 21:57:11 -07:00
# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
2022-02-19 21:57:11 -07:00
# COPY ./app /code/app
## CPU BUILD
FROM python:3.8 as cpu
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6 -y
WORKDIR /code
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
COPY ./app /code/app