automatically set env vars for single gpu deepspeed zero3 (#3118) [skip ci]

* automatically set env vars for single gpu deepspeed zero3

* use setdefault
This commit is contained in:
Wing Lian
2025-08-29 13:36:47 -04:00
committed by GitHub
parent 5b6ec2820f
commit 7ed40f1d70
2 changed files with 7 additions and 9 deletions

View File

@@ -63,15 +63,6 @@ Start from Stage 1 -> Stage 2 -> Stage 3.
:::
::: {.callout-tip}
Using ZeRO Stage 3 with Single-GPU training
ZeRO Stage 3 can be used for training on a single GPU by manually setting the environment variables:
`WORLD_SIZE=1 LOCAL_RANK=0 MASTER_ADDR=0.0.0.0 MASTER_PORT=29500`
:::
## Fully Sharded Data Parallel (FSDP) {#sec-fsdp}
::: {.callout-note}