ResNeXt.pytorch icon indicating copy to clipboard operation
ResNeXt.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon prlz77

GPU memory usage during training

Open realwecan opened this issue 7 years ago • 1 comments

It seems to me that each image uses ~5GB of GPU memory (ResNeXt-152), making it only possible to train with 2 images per GPU (TITAN X). Is that normal? I would appreciate if someone could be able to point out where I can start debugging for this?

realwecan avatar May 22 '18 02:05 realwecan

I have never tried with 152 layers, but it seems normal that memory blows up when using so many layers. Did you find any possible answer?

prlz77 avatar May 24 '18 16:05 prlz77