Questions about using multiple GPU

[Meng-bin]

Dear community,

I have a question about DeepONets use.
One machine has multiple GPU, how do I get a DeepONet to run on multiple GPU? thank you.
I'm using tensorflow.compat.v1

[lululxvi]

https://deepxde.readthedocs.io/en/latest/user/parallel.html