search

MenghaoGuo / AutoDeeplab

Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation
https://arxiv.org/abs/1901.02985
410 stars 97 forks source link

Use GPU '0,1,2,3', but only the card 0 is actually used. #24

Open Wangzhuoying0716 opened 5 years ago

Wangzhuoying0716 commented 5 years ago

Thanks for your contributions! When I tried to run the train_voc.sh on cityscapes dataset only by changing the '--dataset pascal' to '--dataset cityscapes', it turns out that 4 GPUs are assigned but only the first one was actually used. I check the code but can't find the reasons. Could you please give a help? Thanks!

HankKung commented 5 years ago

Same problem

Randylcy commented 5 years ago

Same problem! and I have spent a lot of time to solve that.

YANGWAGN commented 5 years ago

The same problem has been bothering me for a long time.

ghost commented 5 years ago

Same problem.

YANGWAGN commented 5 years ago

@HankKung 请问您后来解决此问题了吗？谢谢！

HankKung commented 5 years ago

@HankKung 请问您后来解决此问题了吗？谢谢！

I haven't worked on this part yet. I've been running on a single GPU currently. Does someone know how to make it?

2793145003 commented 5 years ago

See https://github.com/NoamRosenberg/AutoML which can use all GPUs.