Can't install

[y4my4my4m]

    ~/Software/Direct3D-S2    main  pyenv local 3.12.3                                                                                                                                                                                                                        ✔ 
    ~/Software/Direct3D-S2    main ?1  pip install -r requirements.txt                                                                                                                                                                                             ✔  3.12.3  

Collecting git+https://github.com/mit-han-lab/torchsparse.git (from -r requirements.txt (line 17))
  Cloning https://github.com/mit-han-lab/torchsparse.git to /tmp/pip-req-build-u3izwgfu
  Running command git clone --filter=blob:none --quiet https://github.com/mit-han-lab/torchsparse.git /tmp/pip-req-build-u3izwgfu
  Resolved https://github.com/mit-han-lab/torchsparse.git to commit 385f5ce8718fcae93540511b7f5832f4e71fd835
  Installing build dependencies ... done
  Getting requirements to build wheel ... error
  error: subprocess-exited-with-error
  
  × Getting requirements to build wheel did not run successfully.
  │ exit code: 1
  ╰─> [20 lines of output]
      Traceback (most recent call last):
        File "/home/y4my4m/.pyenv/versions/3.12.3/lib/python3.12/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 353, in <module>
          main()
        File "/home/y4my4m/.pyenv/versions/3.12.3/lib/python3.12/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 335, in main
          json_out['return_val'] = hook(**hook_input['kwargs'])
                                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        File "/home/y4my4m/.pyenv/versions/3.12.3/lib/python3.12/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 118, in get_requires_for_build_wheel
          return hook(config_settings)
                 ^^^^^^^^^^^^^^^^^^^^^
        File "/tmp/pip-build-env-7mbw3cu7/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 331, in get_requires_for_build_wheel
          return self._get_build_requires(config_settings, requirements=[])
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        File "/tmp/pip-build-env-7mbw3cu7/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 301, in _get_build_requires
          self.run_setup()
        File "/tmp/pip-build-env-7mbw3cu7/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 512, in run_setup
          super().run_setup(setup_script=setup_script)
        File "/tmp/pip-build-env-7mbw3cu7/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 317, in run_setup
          exec(code, locals())
        File "<string>", line 4, in <module>
      ModuleNotFoundError: No module named 'torch'
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
error: subprocess-exited-with-error

× Getting requirements to build wheel did not run successfully.
│ exit code: 1
╰─> See above for output.

note: This error originates from a subprocess, and is likely not a problem with pip.

[notice] A new release of pip is available: 24.0 -> 25.1.1
[notice] To update, run: pip install --upgrade pip

Installing torch manually DOES NOT solve the issue...

[SNAKEIX]

yup because the installed method meant for linux and not windows , most people use windows , i don't understand why they choose linux only lol , i hope they add installing method for windows users soon

[y4my4my4m]

@SNAKEIX im on linux, (manjaro KDE)

[SNAKEIX]

i don't know what to say man , i don't want to run the model through WSL2, most open source models are on windows or have on windows and linux

[y4my4my4m]

I don't think there's anything related to being specifically linux or windows in what they've presented so far.

[Linyou]

Thanks for pointing this out. From the error log, it looks like pip tries to install torchsparse before torch. Since torchsparse depends on torch, this causes the installation to fail. We’ll update the requirements.txt file to make sure torch is installed first.

In the meantime, I recommend manually installing torch and torchvision before running pip install -r requirements.txt.

[Linyou]

@y4my4my4m Just to confirm — if you manually install torch first and then install torchsparse, are you still getting the same error? If not, would you mind sharing the updated error log?

[Deathdadev]

Please add --no-build-isolation to the end of your command!

For example:

pip install -r requirements.txt --no-build-isolation

[SNAKEIX]

I don't think there's anything related to being specifically linux or windows in what they've presented so far.

sure because you don't use windows , the problems here is 3 thing:

triton not for windows
flash-attn same thing
udf_ext error

the whole install method should be rebuilt for windows users and test it on windows too, and not trying to figure it out how to make it works while using linux

[Deathdadev]

I don't think there's anything related to being specifically linux or windows in what they've presented so far.

sure because you don't use windows , the problems here is 3 thing:

triton not for windows flash-attn same thing udf_ext error

the whole install method should be rebuilt for windows users and test it on windows too, and not trying to figure it out how to make it works while using linux

flash-attn: https://huggingface.co/lldacing/flash-attention-windows-wheel/tree/main
triton: https://pypi.org/project/triton-windows/
udf_ext: I resolved this issue in my pull request #10

[Stoobs]

torchsparse is now the blocker on windows.

[eadwu]

Here's a Dockerfile that works for me (albeit OOM during runtime)

FROM docker.io/pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel

RUN apt update --assume-yes && \
    apt install --assume-yes \
        git \
        sudo \
        python3 python3-pip \
        libgl1-mesa-glx \
        libglib2.0-0 && \
    rm -rf /var/lib/apt/lists/*

RUN apt update --assume-yes && \
    apt install --assume-yes \
        wget \
        libsparsehash-dev && \
    rm -rf /var/lib/apt/lists/*

RUN <<EOF
wget 'https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl' \
    -O flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
pip install --no-cache-dir flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
rm ./flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
EOF

COPY ./. /direct3d
RUN <<EOF
cd /direct3d
pip install --no-cache-dir --requirement ./requirements.txt
EOF

RUN <<EOF
pip install --no-cache-dir gradio kornia timm trimesh

wget 'https://download.pytorch.org/whl/cu124/xformers-0.0.29.post3-cp311-cp311-manylinux_2_28_x86_64.whl' \
    -O xformers-0.0.29.post3-cp311-cp311-manylinux_2_28_x86_64.whl
pip install --no-cache-dir xformers-0.0.29.post3-cp311-cp311-manylinux_2_28_x86_64.whl
rm ./xformers-0.0.29.post3-cp311-cp311-manylinux_2_28_x86_64.whl
EOF

RUN mkdir -p /workspace/app
WORKDIR /workspace

# ENTRYPOINT ["python", "/app/app.py"]
ENTRYPOINT ["gradio", "/workspace/app/app.py", "--demo-name=demo"]

# podman build --rm --device nvidia.com/gpu=all --tag direct3d:latest -f Dockerfile 

# podman run \
#     --rm -it \
#     --device nvidia.com/gpu=all \
#     --volume ~/.cache/huggingface/hub/models--wushuang98--Direct3D-S2:/root/.cache/huggingface/hub/models--wushuang98--Direct3D-S2:ro \
#     -v "$(pwd)":/workspace/app \
#     -e GRADIO_SERVER_PORT=7860 -e GRADIO_SERVER_NAME=0.0.0.0 \
#     -p 7860:7860 \
#     direct3d:latest

[Deathdadev]

https://github.com/Deathdadev/torchsparse My modified version. Please let others know to test this version to see if it works

[y4my4my4m]

@y4my4my4m Just to confirm — if you manually install torch first and then install torchsparse, are you still getting the same error? If not, would you mind sharing the updated error log?

yes that is correct, manually installing torch and/or torchparse will result in the same error (in fact it'll just keep saying that torch isnt installed despite being installed)

---

Please add --no-build-isolation to the end of your command!

For example:

pip install -r requirements.txt --no-build-isolation

Seems like it goes further within the install but it freezes at
Building wheel for flash-attn (setup.py) ... |

I tried following this post: adithya-s-k/omniparse#36 (comment)
But to no avail.

I have 128gb ram and 24gb vram, don't think my sysetm can't handle building it.

---

@eadwu
Thank you for the dockerfile, i had to modify it but it's working

1. app.py

# add this at the top:
import os
# around line 93

base_dir = os.path.dirname(__file__)
example_dir = os.path.join(base_dir, "assets", "test")
examples = [os.path.join(example_dir, i) for i in os.listdir(example_dir)]

# around line 170
            with gr.Tabs(selected='tab_img_gallery') as gallery:
                with gr.Tab('Image to 3D Gallery', id='tab_img_gallery') as tab_gi:
                    with gr.Row():
                        gr.Examples(
                            examples=examples,
                            inputs=[image_input],
                            label=None, 
                            examples_per_page=24

2. Dockerfile

FROM nvidia/cuda:12.4.1-devel-ubuntu22.04

# Install Python and pip
RUN apt update && apt install -y \
    python3 python3-pip \
    git \
    sudo \
    wget \
    build-essential \
    cmake \
    ninja-build \
    libgl1-mesa-glx \
    libglib2.0-0 \
    libsparsehash-dev && \
    rm -rf /var/lib/apt/lists/*

# Upgrade pip and install PyTorch 2.5.1 for CUDA 12.4
RUN pip install --upgrade pip && \
    pip install torch==2.5.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

# Set architecture for compiling CUDA extensions (e.g. RTX 4090 = compute capability 8.9)
ENV TORCH_CUDA_ARCH_LIST="8.9"

# Copy the app source
COPY . /direct3d

# Install Python dependencies (builds torchsparse, udf_ext, etc.)
RUN cd /direct3d && pip install --no-cache-dir -r requirements.txt

# Install other Python tools and xformers (CUDA 12.4 + PyTorch 2.5.1 compatible)
RUN pip install --no-cache-dir gradio kornia timm trimesh
# Install FlashAttention
RUN wget 'https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl' && \
    pip install --no-cache-dir flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl && \
    rm flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl



# Set working directory and create mount point
RUN mkdir -p /workspace/app
WORKDIR /workspace

# Launch Gradio app
ENTRYPOINT ["gradio", "/workspace/app/app.py", "--demo-name=demo"]

3. docker-compose.yml

services:
  direct3d:
    build:
      context: .
      dockerfile: Dockerfile
    # image: direct3d:latest
    runtime: nvidia
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    environment:
      GRADIO_SERVER_PORT: 7860
      GRADIO_SERVER_NAME: 0.0.0.0
    ports:
      - "7860:7860"
    volumes:
      - ~/.cache/huggingface/hub/models--wushuang98--Direct3D-S2:/root/.cache/huggingface/hub/models--wushuang98--Direct3D-S2
      - ./:/workspace/app
    entrypoint: ["gradio", "/workspace/app/app.py", "--demo-name=demo"]

[y4my4my4m]

Unfortunately, although the above runs and i can open gradio, im getting torchsparse.backend errors when trying to generate, like

dense Sampling:: 100%|██████████| 50/50 [00:05<00:00,  9.45it/s]
sparse512 Sampling:: 100%|██████████| 30/30 [00:13<00:00,  2.22it/s]
direct3d-1  | Traceback (most recent call last):
direct3d-1  |   File "/usr/local/lib/python3.10/dist-packages/gradio/queueing.py", line 625, in process_events
direct3d-1  |     response = await route_utils.call_process_api(
direct3d-1  |   File "/usr/local/lib/python3.10/dist-packages/gradio/route_utils.py", line 322, in call_process_api
direct3d-1  |     output = await app.get_blocks().process_api(
direct3d-1  |   File "/usr/local/lib/python3.10/dist-packages/gradio/blocks.py", line 2193, in process_api
direct3d-1  |     result = await self.call_function(...

direct3d-1 | AttributeError: module 'torchsparse.backend' has no attribute 'build_kernel_map_subm_hashmap'