Pytorch + pytest + multigpu

Hi, I want to test the equality of the gradient on multi-gpus and single-gpu.

The problem is that I don’t have multi-gpu in my local machine.

The question is that could I test the behavior of my code in multi-gpu without multi-gpu environment?

When reminding of my undergrads experience, there is a mockup object that mimics the real behavior to resolve this issue.