Skip to content
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

Out of memory #84

Open
diaodeyi opened this issue Apr 6, 2021 · 2 comments
Open

Out of memory #84

diaodeyi opened this issue Apr 6, 2021 · 2 comments

Comments

@diaodeyi
Copy link

diaodeyi commented Apr 6, 2021

Why the program can run in the training, but stuck in saving weight?

Saving weights to /data1/code/BigGAN-PyTorch/weights/BigGAN_I512_seed0_Gch96_Dch96_bs8_nDa8_nGa8_Glr1.0e-04_Dlr4.0e-04_Gnlinplace_relu_Dnlinplace_relu_Ginitortho_Dinitortho_Gattn64_Dattn64_Gshared_hier_ema...
Saving weights to /data1/code/BigGAN-PyTorch/weights/BigGAN_I512_seed0_Gch96_Dch96_bs8_nDa8_nGa8_Glr1.0e-04_Dlr4.0e-04_Gnlinplace_relu_Dnlinplace_relu_Ginitortho_Dinitortho_Gattn64_Dattn64_Gshared_hier_ema/copy0...

RuntimeError: CUDA out of memory. Tried to allocate 3.75 GiB (GPU 0; 11.91 GiB total capacity; 6.14 GiB already allocated; 416.75 MiB free; 10.94 GiB reserved in total by PyTorch)
@diaodeyi
Copy link
Author

diaodeyi commented Apr 7, 2021

even though the Occupation rate before is only 50%, after the step 1000 ,it still out of memory. I change the code in the utils,
Device('cuda') -> Device('cpu') it seems nothing

@Wanghd-MVP
Copy link

I have the same problem，do you have solved it？
If you have solved it, can you tell me how to solve this problem.
Thank you very much!!!!!!!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@diaodeyi @Wanghd-MVP