New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

CUDA out of memory error #12

Open

jmhjimenez opened this issue Aug 18, 2021 · 0 comments

jmhjimenez commented Aug 18, 2021 •

edited

Hi, when tried this over Colab, I'm encountering the following error with a batch size of 2

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 11.17 GiB total capacity; 10.76 GiB already allocated; 12.81 MiB free; 101.88 MiB cached)
In call to configurable 'train' (<function train at 0x7fb2fb7f1e60>)

There's no error when the batch size is 1. Is there a way to fix this or do I really need a GPU with a higher memory??

For a GPU with 12Gb memory, is the batch size limited to only 1?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment