-
Notifications
You must be signed in to change notification settings - Fork 226
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.
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
Question: I can't release the memory in gpu during execute nt.empirical_ntk_fn #146
Comments
IIUC every for every On a separate note, for CNNs you may want to try Lmk if this helps! |
Thanks!! |
I suggest the |
I have follow the solution in RuntimeError: UNKNOWN: Failed to determine best cudnn convolution algorithm #8506
and High GPU memory during empirical NTK calculation #100
But it doesn't work for me.
This code can reproduce the problem:
And after it run several iteration, the error happen.
The text was updated successfully, but these errors were encountered: