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.

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

CUDA out of memory #5

Open
czgst opened this issue Apr 7, 2024 · 0 comments
Open

CUDA out of memory #5

czgst opened this issue Apr 7, 2024 · 0 comments

Comments

@czgst
Copy link

czgst commented Apr 7, 2024

Thank you for your interesting work and contribution to the research community!
I am trying to evaluate the DSEC dataset with full size of image (480×640) and batch size of 1.
I am getting this out of memory error:
RuntimeError: CUDA out of memory. Tried to allocate 228.00 MiB (GPU 0; 10.76 GiB total capacity; 9.00 GiB already allocated; 7.69 MiB free; 9.44 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
Q1: My GPU 2080Ti has 12GB of memory, do I need more memory?
Q2: If I want to train on the DSEC dataset with cropped size of image (288×384) and batch size of 6, how much memory do I need?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant