pt.darts icon indicating copy to clipboard operation
pt.darts copied to clipboard

very slow in multi-gpu

Open YoungDav opened this issue 5 years ago • 5 comments

Environmental for requests numpy==1.15.4 graphviz==0.8.4 torch==1.0.0 torchvision==0.2.1 tensorboard==1.13.0 tensorboardX==1.6

when I use 4 gpus(GPU-Util about 4-21%), time will be very very slower than only use 1 gpu(GPU-Util about 30-90%)?

YoungDav avatar Dec 05 '19 09:12 YoungDav

I have the same problem. Have you solved it?

JiyueWang avatar Jan 03 '20 11:01 JiyueWang

I have the same problem. Have you solved it?

no

YoungDav avatar Jan 04 '20 03:01 YoungDav

I faced the same problem and found a solution by using DistributedDataParallel from Nvidia apex.

mk-minchul avatar Mar 26 '20 04:03 mk-minchul

Hi @mk-minchul , could you share the solution of DistributedDataParallel? It would be quite helpful to increase the batch size and speedup search process by multi-gpu. Thank you very much!

luzai avatar Jun 10 '20 06:06 luzai