Under the Mnist-Hogwild framework, how to use multi-gpu computing？

pytorch / examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

https://pytorch.org/examples

BSD 3-Clause "New" or "Revised" License

22.19k stars 9.51k forks source link

Under the Mnist-Hogwild framework, how to use multi-gpu computing？ #795

Open Wang-Zhenxing opened 4 years ago

Wang-Zhenxing commented 4 years ago

When I execute the code example of mnist_hogwild, I find that multiple processes are running parallelly on one gpu. Question: Can multiple processes be executed in parallel on multiple GPUs?

tengerye commented 2 years ago

@Wang-Zhenxing Hi, have you found the answer yet?