CUDA out of memory Runtime Error

#78
by fernis - opened

A few days ago I runned this known DreamBooth Google Colab with my face and I was able to get some results. But yesterday I wanted to retrain it again to make it better (tried using the same photos again), and right now, it throws this out of memory exception:
image.png RuntimeError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU 0; 14.76 GiB total capacity; 12.24 GiB already allocated; 501.75 MiB free; 13.16 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

I was wondering if someone had the same problem and know why is this happening now and not before.
I'm linking my colab to show it to you.
Thanks in advance!

Sign up or log in to comment