Spaces:
Sleeping
Sleeping
File size: 2,389 Bytes
514d439 495255b 514d439 495255b 514d439 3d11649 514d439 3d11649 514d439 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | # A100 / dedicated-GPU Hugging Face Space (Docker SDK).
# Models load once at boot and stay warm; pause the Space to stop billing.
FROM pytorch/pytorch:2.4.1-cuda12.1-cudnn9-runtime
ENV DEBIAN_FRONTEND=noninteractive \
PYOPENGL_PLATFORM=egl \
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True \
SAM3D_DIR=/app/sam-3d-body \
HF_HOME=/app/.cache/huggingface
# System libs for OpenCV / pyrender / video decoding, plus a C++ toolchain.
# build-essential is required because the runtime base ships no compiler and
# detectron2 (a1ce2f9) builds its ops from source against the installed torch.
RUN apt-get update && apt-get install -y --no-install-recommends \
git ffmpeg libgl1 libglib2.0-0 libosmesa6 libegl1 \
build-essential ninja-build \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Clone the model repo (added to sys.path by pipeline/gpu.py via SAM3D_DIR)
RUN git clone https://github.com/facebookresearch/sam-3d-body.git /app/sam-3d-body
# Python deps. detectron2 is pinned to a1ce2f9 and built --no-build-isolation
# --no-deps so it compiles against the torch already in the base image — the
# single most fragile step; the pinned base keeps it reproducible.
COPY requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir -r /app/requirements.txt \
&& pip install --no-cache-dir \
"git+https://github.com/facebookresearch/detectron2.git@a1ce2f9" --no-build-isolation --no-deps \
&& pip install --no-cache-dir "git+https://github.com/microsoft/MoGe.git"
# Runtime cache redirection. HF Spaces runs as a non-root user with HOME=/, so
# anything writing to ~/.cache (torch.hub clones DINOv3 here at model load,
# matplotlib, fontconfig, ...) hits a read-only /.cache. These are only needed at
# runtime, so they sit BELOW the pip layer to keep that expensive layer cached.
ENV HOME=/app \
XDG_CACHE_HOME=/app/.cache \
TORCH_HOME=/app/.cache/torch \
MPLCONFIGDIR=/app/.cache/matplotlib
# App (UI + the pipeline package; GPU code is isolated in pipeline/gpu.py)
COPY app.py /app/app.py
COPY pipeline /app/pipeline
# Writable cache + home (HF Spaces runs as a non-root user). torch.hub needs to
# clone into TORCH_HOME at runtime, so the whole /app tree must be writable.
RUN mkdir -p /app/.cache/huggingface /app/.cache/torch /app/.cache/matplotlib \
&& chmod -R 777 /app
EXPOSE 7860
CMD ["python", "app.py"]
|