VITA-Group / LightGaussian

[NeurIPS 2024 Spotlight]"LightGaussian: Unbounded 3D Gaussian Compression with 15x Reduction and 200+ FPS", Zhiwen Fan, Kevin Wang, Kairun Wen, Zehao Zhu, Dejia Xu, Zhangyang Wang
https://lightgaussian.github.io/
Other
608 stars 57 forks source link

nvrtc: error: invalid value for --gpu-architecture (-arch) #22

Open yang1hu opened 8 months ago

yang1hu commented 8 months ago

when I try to train my dataset with train_densify_prune.py ,I encounter the error when the processing to 60% . my machine is unbuntu 20.04 and RTX 4090 GPU i execute command is python train_densify_prune.py -s /root/autodl-tmp/script/test_light --prune_percent 0.6 --prune_decay 0.6 --prune_iterations 20000 --v_pow 0.1 --eval question I saw the similar issue,but i think they are different would you like to help me.

blackbird2003 commented 4 months ago

Hello, I encountered the same problem, and finally I changed the environment to RTX3090 to solve it.