diff --git a/pytorch-ddp-example/requirements.txt b/pytorch-ddp-example/requirements.txt index d506f2e9b88fea2cd534651a7d0e1a8110a9be6b..ca4efd1bc65d46da1166b16fc326adfd788dc959 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.4 +torchrun_jsc>=0.0.5 torchvision>=0.13 diff --git a/pytorch-fsdp-example/requirements.txt b/pytorch-fsdp-example/requirements.txt index a6804188b1c16bd7d189e0665545d0e22e6f8de0..68da8ed1da6a07f964521b1df1b7b7b2e0a9317f 100644 --- a/pytorch-fsdp-example/requirements.txt +++ b/pytorch-fsdp-example/requirements.txt @@ -1,3 +1,3 @@ torch>=2,<3 -torchrun_jsc>=0.0.4 +torchrun_jsc>=0.0.5 torchvision>=0.13