MrForExample / ComfyUI-3D-Pack

An extensive node suite that enables ComfyUI to process 3D inputs (Mesh & UV Texture, etc) using cutting edge algorithms (3DGS, NeRF, etc.)
MIT License
1.72k stars 165 forks source link

Update prebuilt wheel files for Linux #136

Closed YanWenKun closed 2 months ago

YanWenKun commented 2 months ago

This is a follow-up of #126.

In trying to fix #123, #29, #86, I have built wheel files for Linux with build targets:

TORCH_CUDA_ARCH_LIST="6.1+PTX"

In theory, it should be running fine on Pascal (GTX 1000) and later GPUs.

But edge cases came up and now I have covered more with build targets:

TORCH_CUDA_ARCH_LIST="5.2+PTX;6.0;6.1+PTX;7.5;8.0;8.6;8.9"

Now this should be good enough for GPUs from Maxwell (GTX 900) to Ada Lovelace (GTX 4000) according to gencode list.

YanWenKun commented 2 months ago

And also, for anyone who want to build these wheels, I have wrote the steps in this repo: https://github.com/YanWenKun/ComfyUI-3D-Pack-LinuxWheels

MrForExample commented 2 months ago

Thanks friend, I'll add that link into install guide, cheers 👍