From a1add716bcf220ea58a7f151b9d6f5548e47a925 Mon Sep 17 00:00:00 2001 From: lilong12 Date: Fri, 27 Nov 2020 11:44:35 +0800 Subject: [PATCH] Add a flag to control whether to initialize gloo (#29150) --- python/paddle/distributed/fleet/launch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/distributed/fleet/launch.py b/python/paddle/distributed/fleet/launch.py index fbace6ba1f..a7490f770d 100644 --- a/python/paddle/distributed/fleet/launch.py +++ b/python/paddle/distributed/fleet/launch.py @@ -219,7 +219,7 @@ def launch_collective(args): global_envs = copy.copy(os.environ.copy()) gloo_rendezvous_dir = tempfile.mkdtemp() # add gloo env - global_envs["PADDLE_WITH_GLOO"] = "1" + global_envs["PADDLE_WITH_GLOO"] = str(os.getenv("PADDLE_WITH_GLOO", "1")) global_envs["PADDLE_GLOO_RENDEZVOUS"] = "3" global_envs["PADDLE_GLOO_FS_PATH"] = gloo_rendezvous_dir -- GitLab