CUDA out of memory on A6000 [HiFA]

threestudio-project / threestudio

A unified framework for 3D content generation.

Apache License 2.0

5.91k stars 457 forks source link

CUDA out of memory on A6000 [HiFA] #419

Open nkundtz opened 5 months ago

nkundtz commented 5 months ago

I have been getting CUDA out of memory errors when running the HiFA configs. With hifa.yaml it happens almost immediately (iteration 5 or so) with prolificdreamer-hifa.yaml it happened after 12903 iterations.

vilo9595 commented 4 months ago

half num_samples_per_ray in the config/hifa.yaml thought it was an error as well when running on an A100, but with the number set to 256 it needs about 35GB of VRAM