Error when training with multiple GPUs in TAO

Version 22.05 also works with 8 GPUs. Many thanks for the help!

I will follow up on the topic because of the bad performance.