CUDA out of memory error during train_large

DekuLiuTesla / CityGaussian

CityGaussian Series for High-quality Large-Scale Scene Reconstruction with Gaussians

https://dekuliutesla.github.io/CityGaussianV2

Other

530 stars 38 forks source link

Closed Richardder closed 2 months ago

Richardder commented 2 months ago

My GPU is RTX 3090, when I run the train_large.py, it always fails. Should I use GPU with 40G or larger GPU memory? 截屏2024-09-25 22 16 57

ArSpi commented 2 months ago

Please check #9.

Richardder commented 2 months ago

Please check #9.

I tried to set the max_cache_num=256 and resolution=4, it still didn't work.

Richardder commented 2 months ago

It turns out that it's the RAM overflow, not the GPU memory.