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 Issue #16

Open
zy937085318 opened this issue Feb 2, 2018 · 2 comments
Open

Out of Memory Issue #16

zy937085318 opened this issue Feb 2, 2018 · 2 comments

Comments

@zy937085318
Copy link

Do any one meet the error that: RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1511320797808/work/torch/lib/THC/generic/THCStorage.cu:66 I just simply apply this model on titanX and also on my own 1080ti, same issue appeared.
@NicWickman
Copy link

Lower your batch size to 1 and see if it persists, you're probably just running out of VRAM on the hungry networks.

@fantasyzl
Copy link

How long did it take to run the program?

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
3 participants
@NicWickman @zy937085318 @fantasyzl