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.

Sign up for GitHub

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

Jump to bottom

How to reduce the GPU memory usage? #2509

Closed
sandrawing opened this issue Oct 15, 2019 · 2 comments
Closed

How to reduce the GPU memory usage? #2509

sandrawing opened this issue Oct 15, 2019 · 2 comments
Assignees
@huanzhang12
Labels
question

Comments

@sandrawing
Copy link

How can I reduce the GPU memory usage in case that my CPU memory is enough and GPU memory is limited?
Besides, what's the function of GPU and CPU when running tree model in LightGBM? Which part of the tree model computation do GPU and CPU process?
@StrikerRUS
Copy link
Collaborator

Refer to Huan Zhang, Si Si and Cho-Jui Hsieh. "GPU Acceleration for Large-scale Tree Boosting". SysML Conference, 2018. and #768 (comment).

Maybe @huanzhang12 can add something.

@StrikerRUS
Copy link
Collaborator

Please refer to #2657 (comment).

@lock lock bot locked as resolved and limited conversation to collaborators Mar 10, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees

@huanzhang12 huanzhang12

Labels
question
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@huanzhang12 @StrikerRUS @sandrawing