Dockerfile

# Copyright (c) 2021, NVIDIA CORPORATION.  All rights reserved.
#
# NVIDIA CORPORATION and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto.  Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION is strictly prohibited.

FROM nvcr.io/nvidia/pytorch:20.12-py3

ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1

RUN apt install -y git 

RUN pip install imageio-ffmpeg==0.4.3 pyspng==0.1.0

WORKDIR /exp/domain-expansion

COPY ./requirements.txt /exp/domain-expansion/requirements.txt
RUN python -m pip install --no-cache-dir --upgrade -r /exp/domain-expansion/requirements.txt

RUN python -m pip uninstall -y gradio
RUN python -m pip install -U pip setuptools wheel
RUN python -m pip install gradio

# Unset TORCH_CUDA_ARCH_LIST and exec.  This makes pytorch run-time
# extension builds significantly faster as we only compile for the
# currently active GPU configuration.
RUN (printf '#!/bin/bash\nunset TORCH_CUDA_ARCH_LIST\nexec \"$@\"\n' >> /entry.sh) && chmod a+x /entry.sh
ENTRYPOINT ["/entry.sh"]

COPY . .
CMD ["uvicorn", "app:main", "--host", "0.0.0.0", "--port", "7860"]