How to change the code if I wanna use two GPU

[graphic-zhang]

One GPU will out of memory , and I have two GPU in my computer. Could someone kindly tell me how to change the code to use two GPU?
I have tried DataParallel, but GaussianModel is not inherit from nn.Module
Forgive me, this question may be foolish

[uy-seng]

having the same issue, did you get it resolved?

[yangwen-1102]

as I understood, if you have multi-GPU you can assign the submodule to another device, e.g. frontend/backend/viewer don't need to be in one device