Spaces:
Running
Running
| FROM nvidia/cuda:12.1.1-devel-ubuntu22.04 | |
| ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 PYTHONDONTWRITEBYTECODE=1 \ | |
| CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST="8.0" FORCE_CUDA=1 MAX_JOBS=2 | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev libffi-dev liblzma-dev libncursesw5-dev libgdbm-dev uuid-dev build-essential git git-lfs wget ca-certificates \ | |
| libegl1 libgl1 libglib2.0-0 libxi6 libxkbcommon-x11-0 libxrender1 libsm6 libxext6 \ | |
| && rm -rf /var/lib/apt/lists/* | |
| RUN wget -q https://www.python.org/ftp/python/3.11.16/Python-3.11.16.tgz -O /tmp/python.tgz \ | |
| && echo "6c0bd76ab0ec7d94ed400b1497f01ac6c7751c8822615ee0855a3eb2d893ea76 /tmp/python.tgz" | sha256sum -c - \ | |
| && tar -xzf /tmp/python.tgz -C /tmp \ | |
| && cd /tmp/Python-3.11.16 && ./configure --prefix=/opt/python --with-ensurepip=install \ | |
| && make -j4 && make install \ | |
| && /opt/python/bin/python3.11 -m venv /opt/venv \ | |
| && rm -rf /tmp/Python-3.11.16 /tmp/python.tgz | |
| ENV PATH=/opt/venv/bin:/usr/local/cuda/bin:$PATH | |
| WORKDIR /opt/space | |
| RUN python -m pip install --no-cache-dir pip==24.3.1 setuptools==75.6.0 wheel==0.45.1 ninja==1.11.1.3 | |
| COPY docker-requirements.txt . | |
| RUN python -m pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu121 -r docker-requirements.txt | |
| # Build required CUDA binaries from pinned official sources. No data.pyg.org | |
| # access, no unpinned fallback, and no compilation or pip at startup. | |
| RUN python -m pip wheel --no-cache-dir --no-deps --no-build-isolation --no-binary=torch-cluster \ | |
| torch-cluster==1.6.3 -w /opt/wheels \ | |
| && python -m pip install --no-index --no-deps /opt/wheels/torch_cluster-*.whl | |
| RUN git clone --depth 1 --branch v0.7.7 https://github.com/facebookresearch/pytorch3d.git /opt/pytorch3d \ | |
| && python -m pip wheel --no-cache-dir --no-deps --no-build-isolation /opt/pytorch3d -w /opt/wheels \ | |
| && python -m pip install --no-index --no-deps /opt/wheels/pytorch3d-*.whl | |
| RUN python -c "import bpy,torch,torch_cluster,pytorch3d; from pytorch3d import _C; assert torch.__version__ == '2.1.2+cu121'; assert torch.version.cuda == '12.1'; assert torch.ops.torch_cluster.cuda_version() > 0; print('BUILD_BINARY_CHECK_PASS', bpy.app.version_string, torch.__version__, torch_cluster.__version__, pytorch3d.__version__)" | |
| # Keep the exact upstream revision already used by this Space. | |
| RUN git clone https://github.com/jasongzy/Make-It-Animatable.git /opt/space/upstream \ | |
| && git -C upstream checkout d60cc7e01ff8da46448e458dbf450e8967b34e77 \ | |
| && git -C upstream submodule update --init --recursive \ | |
| && wget -q https://github.com/facebookincubator/FBX2glTF/releases/download/v0.9.7/FBX2glTF-linux-x64 -O upstream/util/FBX2glTF \ | |
| && chmod +x upstream/util/FBX2glTF \ | |
| && upstream/util/FBX2glTF --help > /dev/null | |
| COPY docker_patch.py runtime.py preflight.py docker_app.py test_preflight.py ./ | |
| RUN python docker_patch.py \ | |
| && python -m unittest -v test_preflight \ | |
| && python -m compileall -q runtime.py preflight.py docker_app.py upstream \ | |
| && python -m pip freeze > /opt/space/build-packages.txt | |
| RUN useradd -m -u 1000 user && chown -R user:user /opt/space | |
| USER user | |
| ENV HOME=/home/user | |
| EXPOSE 7860 | |
| CMD ["python", "/opt/space/docker_app.py"] | |