"docs/git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "11a9f5ea309c63a50861186679d04aec1f7aad72"
Unverified Commit e69b1eec authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

Turn off multi-node launch if only 1 node (#322)

* turn off multi-node launch if only 1 node
parent de0523d4
...@@ -247,9 +247,6 @@ def main(args=None): ...@@ -247,9 +247,6 @@ def main(args=None):
args.include, args.include,
args.exclude) args.exclude)
if multi_node_exec and not shutil.which('pdsh'):
raise RuntimeError("pdsh is not installed, unable to proceed")
env = os.environ.copy() env = os.environ.copy()
if not args.master_addr: if not args.master_addr:
...@@ -278,6 +275,11 @@ def main(args=None): ...@@ -278,6 +275,11 @@ def main(args=None):
# encode world info as base64 to make it easier to pass via command line # encode world info as base64 to make it easier to pass via command line
world_info_base64 = encode_world_info(active_resources) world_info_base64 = encode_world_info(active_resources)
multi_node_exec = len(active_resources) > 1
if multi_node_exec and not shutil.which('pdsh'):
raise RuntimeError("pdsh is not installed, unable to proceed")
if not multi_node_exec: if not multi_node_exec:
deepspeed_launch = [ deepspeed_launch = [
sys.executable, sys.executable,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment