how to train using multiple gpus?

[PeeInTheWind]

hi, dear authors, thank you for sharing you cool model. i am trying to train the model on my datasets to figure it out how it works well.
the problem is that although it can run with 1 gpu (A-100-unique), the gpu capactiy will soon be filled and the train goes very slowly. so i change to 2 gpus, but an error is reported and i cannot fix it. could you please give me some suggestions how to solve it ?

thanks a lot

[peterdudfield]

Could you give a bit more of the error message and what version of the packages you are running please?

[rutkovskii]

@PeeInTheWind Hi, did you get it to work?