Hi, how to use multi gpus?
louielu1027 opened this issue · 2 comments
louielu1027 commented
I use 8 gpus for infenrence, but it is slowly than 1 gpu......
I changed 'gpu_use = 0' to 'gpu_use = 0,1,2,3,4,5,6,7' in retinanet_inference_example.py.
Is it right?
ZFTurbo commented
No, this variable only controls which GPU use for calculation. Code is for single GPU only. If you need inference on several GPU, just split data in N parts and run inference in parallel on each of them.
louielu1027 commented
Ok, Thanks!