How to utilize multi-gpu

liuruijin17 / LSTR

This is an official repository of End-to-end Lane Shape Prediction with Transformers.

BSD 3-Clause "New" or "Revised" License

647 stars 130 forks source link

Open cmiller2air opened 3 years ago

cmiller2air commented 3 years ago

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 commented 3 years ago

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

Lu-Chengyu commented 3 years ago

I can't wait to see it. :p

Starboy-at-earth commented 3 years ago

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

liuruijin17 commented 3 years ago

@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 commented 3 years ago

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.