set default NUM_WORKERS to 0

deepmodeling · Oct 15, 2024 · 7821997 · 7821997
1 parent 8a59a53
commit 7821997
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pd/utils/env.py b/deepmd/pd/utils/env.py
@@ -23,7 +23,7 @@
     ncpus = len(os.sched_getaffinity(0))
 except AttributeError:
     ncpus = os.cpu_count()
-NUM_WORKERS = int(os.environ.get("NUM_WORKERS", min(4, ncpus)))
+NUM_WORKERS = int(os.environ.get("NUM_WORKERS", min(0, ncpus)))
 # Make sure DDP uses correct device if applicable
 LOCAL_RANK = os.environ.get("LOCAL_RANK", None) or paddle.device.get_device()
 LOCAL_RANK = int(0 if LOCAL_RANK is None else paddle.distributed.get_rank())