Run Pyro on multiple GPUs

Misc.
1

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

1 Like
2

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

3

Hi , I can’t use pyro.plate in multiple GPUs (DataParallel) , anything wrong ?

4

Working on this as part of FR distributed training via Horovod · Issue #2570 · pyro-ppl/pyro · GitHub

2 Likes