Skip to content
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

remove fixed compute capabilities list #24

Closed
wants to merge 1 commit into from
Closed

remove fixed compute capabilities list #24

wants to merge 1 commit into from

Conversation

wanzhenchn
Copy link

Removed fixed compute_capabilities = {80, 86, 89, 90} from setup.py when installing from source code.

It has been got by torch.cuda.get_device_capability() at runtime.

@casper-hansen
Copy link
Owner

@wanzhenchn Thank you for your contribution. Unfortunately, the setup.py will fail to build on some PCs without these compute capabilities specified, and GitHub workflows that build the pip package will also fail. See more in #13

As for the actual compute capabilities, {80, 86, 89, 90} is correct because 80 is the first version of Ampere and the AWQ CUDA kernel supports Ampere and later.

I will have to close this as it would cause errors when building.

@wanzhenchn
Copy link
Author

wanzhenchn commented Sep 5, 2023

@casper-hansen

When I installed it from source code on my single A100 GPUs with cuda 11.7, it only support compute_capabilities = 80, 86,
the 89 & 90 is not supported.

image

The issue is caused by a lower version of CUDA ?

If I removed the 89 & 90, everything is Ok.

@casper-hansen
Copy link
Owner

Unsupported gpu architecture

This error is most likely related to your installation because NVCC does not support compute capability 89 and 90 before CUDA 11.8

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.

2 participants