ITADN

[Bug] Can't run distributed training on multiple nodes with slurm using the oumi distributed wrapper.

#2007ClosedAliliRayane 创建于 2025-11-12
bug
A
AliliRayanecommented
### What happened? When trying to run a training using oumi distrbuted torchrun wrapper with slurm, I wait an endless time with nothing happening (probably a rendez-vous problem). I fixed the rendez-vous endpoint but still not working. ### Steps to reproduce the bug Steps to reproduce : 1. salloc 2 or more nodes using slurm. 2. srun one task per node using the torchrun wrapper (oumi distributed). 3. Wait ### System Info ```shell ____ _ _ __ __ _____ / __ \| | | | \/ |_ _| | | | | | | | \ / | | | | | | | | | | |\/| | | | | |__| | |__| | | | |_| |_ \____/ \____/|_| |_|_____| ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ Oumi environment information: ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ ┌────────────────┬──────────────────────────────────────────────┐ │ Oumi version │ 0.4.2 │ │ Python version │ 3.10.15 │ │ Platform │ Linux-6.8.0-84-generic-x86_64-with-glibc2.39 │ └────────────────┴──────────────────────────────────────────────┘ ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ Installed dependencies: ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ ┏━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┓ ┃ PACKAGE ┃ VERSION ┃ ┡━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━┩ │ accelerate │ 1.11.0 │ │ aiohttp │ 3.13.2 │ │ bitsandbytes │ <not installed> │ │ datasets │ 3.2.0 │ │ diffusers │ <not installed> │ │ einops │ <not installed> │ │ jsonlines │ 4.0.0 │ │ liger-kernel │ 0.6.3 │ │ llama-cpp-python │ <not installed> │ │ lm-eval │ 0.4.9.1 │ │ mlflow │ 3.6.0 │ │ numpy │ 2.2.6 │ │ nvidia-ml-py │ <not installed> │ │ omegaconf │ 2.4.0.dev3 │ │ open_clip_torch │ <not installed> │ │ pandas │ 2.3.3 │ │ peft │ 0.17.1 │ │ pexpect │ 4.9.0 │ │ pillow │ 11.3.0 │ │ pydantic │ 2.11.10 │ │ responses │ 0.25.8 │ │ sglang │ <not installed> │ │ skypilot │ 0.10.5 │ │ tensorboard │ 2.20.0 │ │ timm │ <not installed> │ │ torch │ 2.8.0 │ │ torchdata │ 0.9.0 │ │ torchvision │ 0.23.0 │ │ tqdm │ 4.67.1 │ │ transformers │ 4.57.1 │ │ trl │ 0.21.0 │ │ typer │ 0.20.0 │ │ vllm │ <not installed> │ │ wandb │ 0.21.4 │ └──────────────────┴─────────────────┘ ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ Environment variables: ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ ┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┓ ┃ VARIABLE ┃ VALUE ┃ ┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━┩ │ ACCELERATE_DYNAMO_BACKEND │ <not set> │ │ ACCELERATE_DYNAMO_MODE │ <not set> │ │ ACCELERATE_DYNAMO_USE_DYNAMIC │ <not set> │ │ ACCELERATE_DYNAMO_USE_FULLGRAPH │ <not set> │ │ ACCELERATE_USE_FSDP │ <not set> │ │ CUDA_VISIBLE_DEVICES │ 0,1,2,3,4,5,6,7 │ │ LOCAL_RANK │ <not set> │ │ LOCAL_WORLD_SIZE │ <not set> │ │ OUMI_EXTRA_DEPS_FILE │ <not set> │ │ OUMI_FORCE_EDITABLE_INSTALL │ <not set> │ │ OUMI_SLURM_CONNECTIONS │ <not set> │ │ OUMI_USE_SPOT_VM │ <not set> │ │ RANK │ <not set> │ │ WORLD_SIZE │ <not set> │ └─────────────────────────────────┴─────────────────┘ ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ PyTorch information: ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ ┌────────────────┬───────────────────────┐ │ CUDA available │ True │ │ CUDA version │ 12.8 │ │ cuDNN version │ 91.0.2 │ │ Number of GPUs │ 8 │ │ GPU type │ NVIDIA H100 80GB HBM3 │ │ GPU memory │ 79.2GB │ └────────────────┴───────────────────────┘ ```
关闭于 2025-11-14 3 条评论