Getting-Things-Done-with-Pytorch icon indicating copy to clipboard operation
Getting-Things-Done-with-Pytorch copied to clipboard

CUDA out of memory. Tried to allocate 384.00 MiB (GPU 0; 15.90 GiB total capacity; 14.69 GiB already allocated; 291.75 MiB free; 14.76 GiB reserved in total by PyTorch)

Open abrh119 opened this issue 3 years ago • 2 comments

any fixes for this?

abrh119 avatar Apr 16 '22 11:04 abrh119

Try reducing the batch size.

Alegzandra avatar Dec 07 '22 13:12 Alegzandra