How to use multi-GPUs on a single mechine to run the cases in Modulus

[ngeneva]

Some additional information here:

Testing performance on multiple GPUs

Hi @bvss891 In Modulus the batch size defined in the config are by default the local (per GPU) batch size. What this means is that if you keep the batch size the same between 1 and 2 GPUs is that you’ve gone from a global batch size of 10000 → 20000. Thus this would be a weak scaling test and the best you could ask for is the same performance speed between 1->2 GPUs. This of course won’t happen because to the communication between the GPUs adding some overhead. The exact overhead depends on ho…