MrNeRF / gaussian-splatting-cuda

3D Gaussian Splatting, reimagined: Unleashing unmatched speed with C++ and CUDA from the ground up!
Other
909 stars 74 forks source link

Config to limit CUDA memory usage #21

Closed MrNeRF closed 1 year ago

MrNeRF commented 1 year ago

Discussed in https://github.com/MrNeRF/gaussian-splatting-cuda/discussions/20

Originally posted by **BennetLeff** August 22, 2023 Howdy! I got everything building and running on my 3080ti (so there's a new card confirmed working). The truck demo scene worked fine. Later I tried to import my own scene and torch reserved too much GPU memory. I don't have this problem with the same dataset in the python/original implementation. This project would be more accessible if some configs existed to control this! I don't have time at the moment to fix it but might soon :)

TODO:

Maybe make sure that it works synchronously first. Might be a hustle to get it asynchronous without blocking or introducing bugs. See https://github.com/MrNeRF/gaussian-splatting-cuda/discussions/20

MrNeRF commented 1 year ago

https://github.com/MrNeRF/gaussian-splatting-cuda/pull/23

only loading images asynchronously to gpu left1

MrNeRF commented 1 year ago

Memory Problem should be fixed with --emtpy-gpu-cache flag.