search

DongSky / lbba_boosted_wsod

22 stars 1 forks source link

Problems on GPU and CPU usage #2

Open jwfanDL opened 2 years ago

jwfanDL commented 2 years ago

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 commented 2 years ago

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.