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