fix

mosaicml · Feb 13, 2024 · 66e1205 · 66e1205
1 parent d2b9cd3
commit 66e1205
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/composer/cli/launcher.py b/composer/cli/launcher.py
@@ -291,7 +291,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: