-
Notifications
You must be signed in to change notification settings - Fork 75
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
GPU support on event worker #740
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi,
As discussed with Fahad on discord - the kairon worker is using CPU and not GPU inside the docker container.
I ran some tests to make sure it wasn't a problem on my side;
and
Both worked fine and detected my GPU
I added the same config to
kairon-worker
;but this did not seem to make a difference.
I inquired to check the Dockerfile for the worker, and noticed that there are no packages installed or drivers for the image.
Found some steps here for someone to implement;
https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/efa-start-nccl-base.html
The text was updated successfully, but these errors were encountered: