beidouamg

Hi @sklenard, I am trying to utilizing the multi-GPU feature, but I have some trouble with it. I install the ddp branch with pytorch 2.1.1 by changing `"torch>=1.8,=1.8,