search

graphdeco-inria / gaussian-splatting

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"
https://repo-sam.inria.fr/fungraph/3d-gaussian-splatting/
Other
14.92k stars 1.95k forks source link

out of memory #1052

Open jiangroubao opened 3 weeks ago

jiangroubao commented 3 weeks ago

It seems that this version runs out of GPU memory more easily compared to the previous one. With the same quality and quantity of photos, the previous version could complete the training (30,000 steps), but now this version throws an out-of-memory exception. Is there any way to resolve this?

jaco001 commented 3 weeks ago

In reality you have three options:

  1. refine dataset (some work)
  2. change GPU for more VRAM (possible)
  3. wait for code optimization/change (wait ;)

What you prefer?

AsherJingkongChen commented 3 weeks ago

You can disable reference image preloading by modifying the code. Just make sure to release preallocations from Torch.

This approach should only affect training speed by a little.