8000 Out of memory · Issue #12 · djukicn/loca · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Out of memory #12

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

Open
renaldfredy opened this issue Nov 7, 2024 · 0 comments
Open

Out of memory #12

renaldfredy opened this issue Nov 7, 2024 · 0 comments

Comments

@renaldfredy
Copy link

Hi djukicn

I have tried run this code using google colab, local PC, even server of my lab, but always getting out of memory. "torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.00 GiB. GPU 0 has a total capacity of 10.75 GiB of which 29.81 MiB is free. Including non-PyTorch memory, this process has 10.71 GiB memory in use. Of the allocated memory 10.44 GiB is allocated by PyTorch, and 12.10 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management "

However, when I reduce batch_size, many error came up such as with_emb in ope.py. Do you have any suggesstion for me regarding this issue? thank you in advance.

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
0