Skip to content

Update prebuilt wheel files for Linux #136

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Apr 13, 2024

Conversation

YanWenKun
Copy link
Contributor

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.

With build targets TORCH_CUDA_ARCH_LIST="5.2+PTX;6.0;6.1+PTX;7.5;8.0;8.6;8.9"
@YanWenKun
Copy link
Contributor Author

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 MrForExample merged commit bc3c275 into MrForExample:main Apr 13, 2024
@MrForExample
Copy link
Owner

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

ComfyUI-3D-Pack fails to load in ComfyUI on Linux system
2 participants