search

Ramprasad-Group / polygnn

polyGNN is a Python library to automate ML model training for polymer informatics.
Other
32 stars 5 forks source link

Memory Allocation Error #16

Closed oliverhvidsten closed 1 year ago

oliverhvidsten commented 1 year ago

I got this error. Any thoughts of how to fix?

RuntimeError: CUDA out of memory. Tried to allocate 30.00 MiB (GPU 0; 15.89 GiB total capacity; 2.14 GiB already allocated; 3.31 MiB free; 2.31 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

rishigurnani commented 1 year ago

What happens when you halve the batch size?

rishigurnani commented 1 year ago

@oliverhvidsten what is the status of this issue?

rishigurnani commented 1 year ago

Closing, since there's been a couple of weeks since the last response. Feel free to reopen if needed.