How to setup DDP without SLURM. I have 2 GPUs on one machine and want to run the traning script on a small dataset.

caizhongang / SMPLer-X

Official Code for "SMPLer-X: Scaling Up Expressive Human Pose and Shape Estimation"

https://caizhongang.github.io/projects/SMPLer-X/

Other

1.01k stars 73 forks source link

How to setup DDP without SLURM. I have 2 GPUs on one machine and want to run the traning script on a small dataset. #72

Closed rmheyg closed 3 months ago

rmheyg commented 3 months ago

Hi I have one machine with 2 GPUs and around 5000 frames with GT SMPLX. I wanted to run the training script but I can't figure out the DDP setup. I have set the world_size=2, and used mp.spawn to allocate rank but it is not working

rmheyg commented 3 months ago

Got it