search

DengPingFan / FSGAN

48 stars 10 forks source link

gpu大小 #4

Closed ludandandan closed 1 year ago

ludandandan commented 1 year ago

你好，Our model can only train on GPU >=32 GB at present，请问是必须单卡就有32G吗，我现在有两张24G的4096的卡，训练的时候设置了--gpu_ids 0,1 ,但是发现只用上了0号卡，并且报out of memeory

ZhengPeng7 commented 1 year ago

是的, 24G的貌似确实是没法训... 因为现batchsize已经很小了, 即使多卡, 0号卡的占用显存是不降的.

ludandandan commented 1 year ago

嗷嗷，好的，谢谢回复～