lerobot icon indicating copy to clipboard operation
lerobot copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

2025, Can we support multi-gpu training?

Open lucasjinreal opened this issue 5 months ago • 2 comments

2025, Can we support multi-gpu training?

lucasjinreal avatar Jun 14 '25 03:06 lucasjinreal

Not yet, but we are working on it

pkooij avatar Jun 14 '25 04:06 pkooij

Thanks for the catch up. Kindly remind to be sure to test with the latest torch. Properly handle DTensor issues when using the accelerator. It doesn't work well with the DTensor feature in the newest torch. Some pull request in lerobot actually not well tested.

lucasjinreal avatar Jun 14 '25 04:06 lucasjinreal

MultiGPU support is here!!: https://huggingface.co/docs/lerobot/en/multi_gpu_training

jadechoghari avatar Oct 17 '25 09:10 jadechoghari