CUDA out of memory - Githubissues

zju3dv / NeuralRecon

Code for "NeuralRecon: Real-Time Coherent 3D Reconstruction from Monocular Video", CVPR 2021 oral

https://zju3dv.github.io/neuralrecon/

Apache License 2.0

2.07k stars 299 forks source link

CUDA out of memory #49

Closed tayton42 closed 3 years ago

tayton42 commented 3 years ago

Because my gpu memory is too small, when I run demo.py I get the following error RuntimeError: CUDA out of memory. Tried to allocate 666.00 MiB (GPU 0; 2.00 GiB total capacity; 557.62 MiB already allocated; 549.02 MiB free; 662.00 MiB reserved in total by PyTorch) How do I modify the code so that it can run on 2G memory, thanks!

JiamingSuen commented 3 years ago

It will require a change in voxel sizes in order to fit 2GB GPU memory, which means that you will have to train the network from scratch.