FROM nvidia/cuda

WORKDIR /srv

COPY requires.gpu .

ARG tuna=https://pypi.tuna.tsinghua.edu.cn/simple

RUN pip install --no-cache-dir --upgrade pip -i $tuna && \
 pip install --no-cache-dir -r requires.gpu -i $tuna && \
 sed -i "s@http://deb.debian.org@https://mirrors.tuna.tsinghua.edu.cn@g" /etc/apt/sources.list && \
 apt update && apt install -y python3 --no-install-recommends && \
 apt-get clean && rm -rf /var/lib/apt/lists/*

CMD ["gunicorn", "app:app"]