diff --git a/pytorch-ddp-example/set_up.sh b/pytorch-ddp-example/set_up.sh
index 0e92ab559cc0b835581673e6d0c176ffa5468488..95d837a1b8126783d740b2550babb4627fdf7cd9 100644
--- a/pytorch-ddp-example/set_up.sh
+++ b/pytorch-ddp-example/set_up.sh
@@ -5,7 +5,7 @@ curr_dir="$(dirname "$curr_file")"
 
 if ! [ -d "$curr_dir"/env ]; then
     [ -x "$(command -v deactivate)" ] && deactivate
-    module purge
+    module --force purge
     source "$curr_dir"/modules.sh
     python3 -m venv --system-site-packages "$curr_dir"/env
     source "$curr_dir"/env/bin/activate
diff --git a/pytorch-fsdp-example/set_up.sh b/pytorch-fsdp-example/set_up.sh
index 0e92ab559cc0b835581673e6d0c176ffa5468488..95d837a1b8126783d740b2550babb4627fdf7cd9 100644
--- a/pytorch-fsdp-example/set_up.sh
+++ b/pytorch-fsdp-example/set_up.sh
@@ -5,7 +5,7 @@ curr_dir="$(dirname "$curr_file")"
 
 if ! [ -d "$curr_dir"/env ]; then
     [ -x "$(command -v deactivate)" ] && deactivate
-    module purge
+    module --force purge
     source "$curr_dir"/modules.sh
     python3 -m venv --system-site-packages "$curr_dir"/env
     source "$curr_dir"/env/bin/activate