diff --git a/pytorch-ddp-example/requirements.txt b/pytorch-ddp-example/requirements.txt index ca4efd1bc65d46da1166b16fc326adfd788dc959..a7006b3ab359635392f810f328ef79224d4b787b 100644 --- a/pytorch-ddp-example/requirements.txt +++ b/pytorch-ddp-example/requirements.txt @@ -1,3 +1,3 @@ torch>=1.9,<3 -torchrun_jsc>=0.0.5 +torchrun_jsc>=0.0.15 torchvision>=0.13 diff --git a/pytorch-fsdp-example/requirements.txt b/pytorch-fsdp-example/requirements.txt index 68da8ed1da6a07f964521b1df1b7b7b2e0a9317f..f2b5ced1c67c696aeb67612003ef6278b626b6b2 100644 --- a/pytorch-fsdp-example/requirements.txt +++ b/pytorch-fsdp-example/requirements.txt @@ -1,3 +1,3 @@ torch>=2,<3 -torchrun_jsc>=0.0.5 +torchrun_jsc>=0.0.15 torchvision>=0.13