-
Notifications
You must be signed in to change notification settings - Fork 4.3k
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
[BUG]: docker build cuda extension error #5732
Comments
Title: [BUG]: docker build cuda extension error |
RUN BUILD_EXT=1 pip install colossalai-nightly: |
one run in container with gpu device param |
workaround Install ColossalAI from a specific commitARG VERSION=main |
thanks |
thanks |
This is because docker buildkit is not compatible with current cuda extension. You can set |
Is there an existing issue for this bug?
🐛 Describe the bug
when docker build run follow command
RUN BUILD_EXT=1 pip install colossalai-nightly
RuntimeError: [extension] Could not find any kernel compatible with the current environment.
but if I run this command in a container (with gpu flag to use GPU cards) then It suceed
base image
FROM nvcr.io/nvidia/cuda:11.8.0-devel-ubuntu20.04
Environment
No response
The text was updated successfully, but these errors were encountered: