Multi-GPU Training?

Question

Multi-GPU Training?

toomanyqs opened this issue 7 years ago · comments

Does this project not have Multi-GPU support? I have been trying to train Deepmask and Sharpmask on my own dataset on an Amazon EC2 p2.16xlarge instance, but it is only using one of the K80s instead of all 16.

nico00 · Answer 1 · Thu Jan 26 2017 08:09:30 GMT+0800 (China Standard Time)

Did you try with -gpu 16 ?

toomanyqs · Answer 2 · Mon Jan 30 2017 20:37:02 GMT+0800 (China Standard Time)

-gpu 'n' uses nth gpu rather than using n gpus.

Kevin Casado · Answer 3 · Tue Feb 28 2017 04:16:20 GMT+0800 (China Standard Time)

Any update on this? I am having the same problem.

Shunyu Yao · Answer 4 · Tue Mar 06 2018 21:46:47 GMT+0800 (China Standard Time)

I noticed it also. Maybe need to use something like nn parallel for multi gpu support.