BEVDepth icon indicating copy to clipboard operation
BEVDepth copied to clipboard

Published 20 hours ago • verified publisher icon Megvii-BaseDetection

multigpu training

Open Co4AI opened this issue 2 years ago • 2 comments

Thanks for your excellent work!

I use this command to train net: python [EXP_PATH] --amp_backend native -b 8 --gpus 8 But only gpu 0 works. How to use multiple gpus to train net.

Co4AI avatar Sep 02 '22 12:09 Co4AI

This should work without any extra settings, please check your local environment.

yinchimaoliang avatar Sep 03 '22 05:09 yinchimaoliang

Thanks for your excellent work!

I use this command to train net: python [EXP_PATH] --amp_backend native -b 8 --gpus 8 But only gpu 0 works. How to use multiple gpus to train net.

meet the same error. this help me fix it. https://github.com/Lightning-AI/lightning/issues/5254#issuecomment-1177602254

zhangjinsong3 avatar Oct 12 '22 08:10 zhangjinsong3