How to change the default gpu of diff-gaussian-rasterization

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

13.63k stars 1.76k forks source link

How to change the default gpu of diff-gaussian-rasterization #933

Open Zazexy opened 1 month ago

Zazexy commented 1 month ago

I wanna use the diff-gaussian-rasterization in other gpu at the same time to process batch data. However, the rasterizer work only on gpu:0， how can I change it.

zhangseammm commented 3 weeks ago

you can set env variable to specify a GPU. Such as: export CUDA_VISIBLE_DEVIDES=1

Zazexy commented 3 weeks ago

Thanks for your reply. However, what I wanna know is how to use multiple gpus at the same time. It doesn't mean that training one scene on different gpus. I just wanna train different scenes on different gpus with the same code.