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

Fixed CUDA out of memory #108

Open

jpramosi wants to merge 2 commits into Anjok07:v5-beta-cml from jpramosi:v5-beta-cml-dev0

jpramosi commented Sep 21, 2021 •

edited

This change will clear the torch cache on each iteration and avoids errors like these:

RuntimeError: CUDA out of memory. Tried to allocate 288.00 MiB (GPU 0; 3.94 GiB total capacity; 793.31 MiB already allocated; 1.89 GiB free; 1008.73 MiB allowed; 936.00 MiB reserved in total by PyTorch)

reapler added 2 commits

September 21, 2021 04:52


          Fixed CUDA out of memory

afc27e6


          Add missing cuda check

a8ff575

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