openclimatefix/skillful_nowcasting

how to train using multiple gpus?

Opened this issue · 2 comments

PeeInTheWind commented

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

image

peterdudfield commented

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

rutkovskii commented

@PeeInTheWind Hi, did you get it to work?