From 11e6143f47de6000405d04fd5055b3b33c6fb1e5 Mon Sep 17 00:00:00 2001 From: Thomas Viehmann Date: Mon, 27 May 2024 12:10:56 +0200 Subject: [PATCH] disable calling torchrun in the zero to thunder notebook (#452) --- notebooks/zero_to_thunder.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/notebooks/zero_to_thunder.ipynb b/notebooks/zero_to_thunder.ipynb index c8d27a89ce..aec01677d0 100644 --- a/notebooks/zero_to_thunder.ipynb +++ b/notebooks/zero_to_thunder.ipynb @@ -3389,7 +3389,8 @@ } ], "source": [ - "!torchrun --nproc_per_node=2 zero_to_thunder_fsdp_simple_example.py" + "# commented out for CI limitations, see https://github.com/Lightning-AI/lightning-thunder/issues/465\n", + "# !torchrun --standalone --nnodes=1 --nproc_per_node=2 zero_to_thunder_fsdp_simple_example.py" ] }, {