dream
1# syntax=docker/dockerfile:experimental
2
3FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime
4
5WORKDIR /src
6
7ARG PRETRAINED_MODEL_NAME_OR_PATH
8ENV PRETRAINED_MODEL_NAME_OR_PATH ${PRETRAINED_MODEL_NAME_OR_PATH}
9ARG SERVICE_PORT
10ENV SERVICE_PORT ${SERVICE_PORT}
11
12COPY ./requirements.txt /src/requirements.txt
13RUN pip install -r /src/requirements.txt
14
15COPY . /src
16
17HEALTHCHECK --interval=5s --timeout=90s --retries=3 CMD curl --fail 127.0.0.1:${SERVICE_PORT}/healthcheck || exit 1
18
19
20CMD gunicorn --workers=1 server:app -b 0.0.0.0:${SERVICE_PORT} --timeout=1200
21