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

Problems on GPU and CPU usage #2

Open
jwfanDL opened this issue Jan 29, 2022 · 1 comment
Open

Problems on GPU and CPU usage #2

jwfanDL opened this issue Jan 29, 2022 · 1 comment

Comments

@jwfanDL
Copy link

jwfanDL commented Jan 29, 2022

Hi~ I meet some problems when training the model
I have set CUDA_VISIBEL_DEVICES=0,1,2,3, however, the code may just use gpu0.
At the same time, cpu usage rate is quite high.

So how can I use more GPU to train the model? I would appreciate if you can answer the question

@DongSky
Copy link
Owner

DongSky commented Feb 2, 2022

Hi JwFbupt
Currently, our LBBA only supports single GPU training, just like the description in readme "GPU with larger memory is better".
And now we are working on migrating the original official code into the multi-GPU detectron2 codebase, but it still needs some time due to some other things.
Best wishes.

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

No branches or pull requests

2 participants