| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:82f38ebced7b2756311fb492d3d44cc131b22654e8620baa93883537a3e355aa in / |
| CMD ["/bin/bash"] |
| ARG PYTORCH_VERSION=2.7.1 |
| ARG TRITON_VERSION= |
| ARG TARGETPLATFORM=linux/amd64 |
| ARG CUDA_VERSION=12.8.0 |
| LABEL com.nvidia.volumes.needed=nvidia_driver |
| RUN |4 PYTORCH_VERSION=2.7.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.8.0 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit |
| COPY /opt/conda /opt/conda # buildkit |
| RUN |4 PYTORCH_VERSION=2.7.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.8.0 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit |
| ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PYTORCH_VERSION=2.7.1 |
| WORKDIR /workspace |
| ARG UPSCALARR_BUILD_VERSION=2026.08.13.8 |
| ENV PYTHONDONTWRITEBYTECODE=1 |
| ENV PYTHONUNBUFFERED=1 |
| ENV PIP_NO_CACHE_DIR=1 |
| ENV UPSCALARR_BUILD_VERSION=2026.08.13.8 |
| ENV CUDA_HOME=/opt/conda |
| ENV TORCH_CUDA_ARCH_LIST=12.0 |
| WORKDIR /app |
| RUN |1 UPSCALARR_BUILD_VERSION=2026.08.13.8 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends libgl1 libglib2.0-0 ffmpeg && rm -rf /var/lib/apt/lists/* # buildkit |
| COPY requirements.txt . # buildkit |
| COPY /wheels/mmcv-2.2.0-*.whl /tmp/ # buildkit |
| RUN |1 UPSCALARR_BUILD_VERSION=2026.08.13.8 /bin/sh -c python -m pip install --upgrade pip setuptools wheel && python -m pip install -r requirements.txt && python -m pip install mmengine==0.10.5 && python -m pip install /tmp/mmcv-2.2.0-*.whl && rm -f /tmp/mmcv-2.2.0-*.whl && python -m pip uninstall -y opencv-python && python -m pip install opencv-python-headless # buildkit |
| ARG RBVSR_CHECKPOINT_URL=https://huggingface.co/akhaliq/RealBasicVSR_x4/resolve/main/RealBasicVSR_x4.pth |
| ADD https://huggingface.co/akhaliq/RealBasicVSR_x4/resolve/main/RealBasicVSR_x4.pth /opt/checkpoints/RealBasicVSR_x4.pth # buildkit |
| ARG BASICVSRPP_CHECKPOINT_URL=https://download.openmmlab.com/mmediting/restorers/basicvsr_plusplus/basicvsr_plusplus_c64n7_8x1_600k_reds4_20210217-db622b2f.pth |
| ADD https://download.openmmlab.com/mmediting/restorers/basicvsr_plusplus/basicvsr_plusplus_c64n7_8x1_600k_reds4_20210217-db622b2f.pth /opt/checkpoints/BasicVSRPP_x4.pth # buildkit |
| ARG VRT_CHECKPOINT_URL=https://github.com/JingyunLiang/VRT/releases/download/v0.0/002_VRT_videosr_bi_REDS_16frames.pth |
| ADD https://github.com/JingyunLiang/VRT/releases/download/v0.0/002_VRT_videosr_bi_REDS_16frames.pth /opt/checkpoints/VRT_x4.pth # buildkit |
| ARG RVRT_CHECKPOINT_URL=https://github.com/JingyunLiang/RVRT/releases/download/v0.0/001_RVRT_videosr_bi_REDS_30frames.pth |
| ADD https://github.com/JingyunLiang/RVRT/releases/download/v0.0/001_RVRT_videosr_bi_REDS_30frames.pth /opt/checkpoints/RVRT_x4.pth # buildkit |
| ARG EDVR_CHECKPOINT_URL=https://download.openmmlab.com/mmediting/restorers/edvr/edvrl_c128b40_8x8_lr2e-4_600k_reds4_20220104-4509865f.pth |
| ADD https://download.openmmlab.com/mmediting/restorers/edvr/edvrl_c128b40_8x8_lr2e-4_600k_reds4_20220104-4509865f.pth /opt/checkpoints/EDVR_x4.pth # buildkit |
| COPY app ./app # buildkit |
| COPY /root/.cache/torch_extensions /root/.cache/torch_extensions # buildkit |
| EXPOSE [8100/tcp] |
| CMD ["uvicorn" "app.main:app" "--host" "0.0.0.0" "--port" "8100"] |
