Update env variable

mosaicml · May 8, 2024 · d6f823d · d6f823d
1 parent 9f7db04
commit d6f823d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/composer/cli/launcher.py b/composer/cli/launcher.py
@@ -339,7 +339,7 @@ def _launch_processes(
             MASTER_ADDR=master_addr,
             MASTER_PORT=str(master_port),
             PYTHONUNBUFFERED='1',
-            NCCL_ASYNC_ERROR_HANDLING='1',
+            TORCH_NCCL_ASYNC_ERROR_HANDLING='1',
         ):
             # Populate the distributed variables in all launcher args
             for arg in training_script_args: