diff --git a/pytorch-ddp-example/requirements.txt b/pytorch-ddp-example/requirements.txt index fae50962424c13e2bd085ad23e29ce2626c1475c..d506f2e9b88fea2cd534651a7d0e1a8110a9be6b 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 -torchvision +torchvision>=0.13 diff --git a/pytorch-ddp-example/run.sbatch b/pytorch-ddp-example/run.sbatch index 6c8ef7a7930533ebc3e0785053d0de59c5ef2b05..68c316553a013fd710eb02594abf53aa493f264b 100644 --- a/pytorch-ddp-example/run.sbatch +++ b/pytorch-ddp-example/run.sbatch @@ -20,7 +20,8 @@ export SRUN_CPUS_PER_TASK="$SLURM_CPUS_PER_TASK" [ -x "$(command -v deactivate)" ] && deactivate -module purge +module --force purge +module load Stages source "$curr_dir"/modules.sh if ! [ -d "$curr_dir"/env ]; then diff --git a/pytorch-fsdp-example/requirements.txt b/pytorch-fsdp-example/requirements.txt index 5283f0337adfe31d5a0fedd756e0229df8bf8209..a6804188b1c16bd7d189e0665545d0e22e6f8de0 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 -torchvision +torchvision>=0.13 diff --git a/pytorch-fsdp-example/run.sbatch b/pytorch-fsdp-example/run.sbatch index 233d8b73119a05ccc984f72eaa311878536a60c0..4af8053e43d773ef0069351d915889b948e6f8a2 100644 --- a/pytorch-fsdp-example/run.sbatch +++ b/pytorch-fsdp-example/run.sbatch @@ -20,7 +20,8 @@ export SRUN_CPUS_PER_TASK="$SLURM_CPUS_PER_TASK" [ -x "$(command -v deactivate)" ] && deactivate -module purge +module --force purge +module load Stages source "$curr_dir"/modules.sh if ! [ -d "$curr_dir"/env ]; then