RuntimeError: CUDA out of memory. Tried to allocate 1.07 GiB (GPU 0; 4.00 GiB total capacity; 2.57 GiB already allocated; 84.45 MiB free; 2.59 GiB reserved in total by PyTorch) about golfdb HOT 5 CLOSED

Hmm. The batch size is 1 by default in eval.py. You could try lowering seq_length. Looks like you only have 4 GB of GPU memory. I used a GPU with 12 GB memory.

from golfdb.

Is there any way to know how big a model or a network my system can handle
without running into this issue?

from golfdb.

Your GPU ran out of memory. Try lowering the batch size.

from golfdb.

yes!. I reduced the batch size to 1 and now it works with train.py.. but the problem now is in running eval.py (is showing the same error) :
"RuntimeError: CUDA out of memory. Tried to allocate 58.00 MiB (GPU 0; 4.00 GiB total capacity; 2.49 GiB already allocated; 44.45 MiB free; 2.57 GiB reserved in total by PyTorch)""..
have you any idea ??

from golfdb.

@wmcnally it works..thank you

from golfdb.