riccorl's picture
first commit
626eca0
raw
history blame
966 Bytes
FROM nvidia/cuda:12.2.0-base-ubuntu20.04
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update \
&& apt-get install \
curl wget python3.10 \
python3.10-distutils \
python3-pip \
curl wget -y \
&& rm -rf /var/lib/apt/lists/*
# FastAPI section
# device env
ENV DEVICE="cuda"
# Copy and install requirements.txt
COPY ./gpu-requirements.txt ./requirements.txt
COPY ./src /app
COPY ./scripts/start.sh /start.sh
COPY ./scripts/gunicorn_conf.py /gunicorn_conf.py
COPY ./scripts/start-reload.sh /start-reload.sh
COPY ./scripts/prestart.sh /app
COPY ./VERSION /
RUN mkdir -p /app/resources/model \
&& pip install --upgrade --no-cache-dir -r requirements.txt \
&& chmod +x /start.sh \
&& chmod +x /start-reload.sh
ARG MODEL_NAME_OR_PATH
WORKDIR /app
ENV PYTHONPATH=/app
EXPOSE 80
# Run the start script, it will check for an /app/prestart.sh script (e.g. for migrations)
# And then will start Gunicorn with Uvicorn
CMD ["/start.sh"]