How to utilize multi-gpu

[cmiller2air]

Hi, I'm trying to train the model with four GPUs by setting batch size : [32] chunk size : [8,8,8,8]

is this correct? it's not working quiet well...

[liuruijin17]

Hi, Yes, the multi-GPU training is not working. I might fix it in a future version. Thanks a lot!

[Lu-Chengyu]

I can't wait to see it. :p

[Starboy-at-earth]

Dear @liuruijin17, Could you please tell me the the device information you used in the training process?

[liuruijin17]

@cmiller2air @Starboy-at-earth Sorry for the months of waiting, I have done the multi-GPU codes and will release them after some testing 🤣

@Starboy-at-earth CPU: AMD Ryzen 1900x GPU: 2x1080Ti DIMM: 4x16 DDR4 3200MHz SSD: 2x2T SAMSUNG 970 EVO Plus 2T This is my ancient working PC🤣

[Starboy-at-earth]

2T prides you....

---Original--- From: @.> Date: Mon, Nov 8, 2021 11:32 AM To: @.>; Cc: @.@.>; Subject: Re: [liuruijin17/LSTR] How to utilize multi-gpu (#35)

@cmiller2air @Starboy-at-earth Sorry for the months of waiting, I have done the multi-GPU codes and will release them after some testing 🤣

@Starboy-at-earth CPU: AMD Ryzen 1900x GPU: 2x1080Ti DIMM: 4x16 DDR4 3200MHz SSD: 2x2T SAMSUNG 970 EVO Plus 2T This is my ancient working PC🤣

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or unsubscribe.