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

Problems with running pre trained models #133

Open
zfkdsg opened this issue Feb 29, 2024 · 1 comment
Open

Problems with running pre trained models #133

zfkdsg opened this issue Feb 29, 2024 · 1 comment

Comments

@zfkdsg
Copy link

zfkdsg commented Feb 29, 2024

I want to observe the effect on my own computer, and I downloaded the logo from pre train. However, I encountered a bug while running the sentence "run nerf. py -- config configs/lego. txt -- rended_only" in Python :
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.11 GiB. GPU 0 has a total capacity of 3.81 GiB of which 664.00 MiB is free. Including non-PyTorch memory, this process has 3.15 GiB memory in use. Of the allocated memory 2.41 GiB is allocated by PyTorch, and 654.21 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 (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)

please help me thanks !!!

@GauravNerf
Copy link

just delete the pycache folder before running the code that might be pre generated from the failed previous training and then restart the render_only command, this should work unless you have very low gpu memory.

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

2 participants