lingvo icon indicating copy to clipboard operation
lingvo copied to clipboard

OOM issue with a tensor of shape [12, 620, 50, 1024] on 8 GPU (12 gb mem each) machine

Open manish-kumar-garg opened this issue 5 years ago • 0 comments

Hi, I am implementing a model in Lingvo and getting OOM with a tensor of shape [12, 620, 50, 1024]. I am using a 8 GPU (Tesla K80, 8 GB each)amazon deep learning AMI with 32 cores and 480 GB RAM. Is this tensor too large for this machine?

manish-kumar-garg avatar Nov 27 '19 11:11 manish-kumar-garg