CUDA out of memory,even set batch_size to 1 · Issue #185 ...
github.com › NVIDIA › waveglowMar 04, 2020 · In Google Colab, with a batch size of 1, it gives out of memory error for an audio 5 seconds long. waveglow = torch.hub.load ('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_waveglow', model_math='fp32') waveglow = waveglow.remove_weightnorm (waveglow) waveglow = waveglow.to ('cuda') waveglow.eval () audio = waveglow.infer (mel.cuda ()) Before the model => GPU Memory = 1100MiB / 15109MiB.