-
Notifications
You must be signed in to change notification settings - Fork 3.8k
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
Windows GPU wheels #3143
Windows GPU wheels #3143
Conversation
…' into windows-gpu-builds-take-2
I realize that Linux world is more diverse, but catboost python library devs manage to achieve this goal also for Linux. It saves users a lot of time... (At the cost of pinning to some Cuda build - ideally there should be versions built for Cuda 8, 9, 10, and 11). |
Whenever you have time, can you please merge in |
…' into windows-gpu-builds-take-3
This pull request has been automatically locked since there has not been any recent activity since it was closed. To start a new related discussion, open a new issue at https://github.com/microsoft/LightGBM/issues including a reference to this. |
Fixes #2263.
This builds on top of #3144, so that'll need to get merged first. Once that is merged, it will be possible to create a single wheel on Windows that work with both CPU and GPU devices. This PR then changes the CI build process to actually use that functionality, so that the standard Windows wheels will work on GPU devices too.
Unlike my previous attempt, the same wheel should work on both GPU and CPU, which makes life easier for everyone, both maintainers (you) and users.
--opencl-python-package
option.