Run Pyro on multiple GPUs

Hello,
is it possible to run Pyro on a machine with multiple GPU cards?
I don’t see any reference to this use-case scenario in the pyro documentation.

Pytorch handles multiple GPU (
https://pytorch.org/tutorials/beginner/former_torchies/parallelism_tutorial.html
so this might be possible in Pyro as well.

Thanks

yes it is the same as in pytorch. do you have a use case that doesn’t work?