eg3d icon indicating copy to clipboard operation
eg3d copied to clipboard
verified publisher icon NVlabs

How much GPU Memory required?

Open KeqiangSun opened this issue 3 years ago • 0 comments

Hi, I try to train this modoel with 12GB GPUs, with "--gpus=8 --batch=8 --mbstd-group=1". But it stuck and I noticed the memory on the first GPU exhausted at a moment. So I wonder:

  1. How much memory does it require to train eg3d with arguments "--gpus=8 --batch=32"?
  2. Are the 16GB V100s okay to train the eg3d?

Thank you!

KeqiangSun avatar Jul 20 '22 03:07 KeqiangSun