diff --git a/video_prediction_tools/env_setup/create_env.sh b/video_prediction_tools/env_setup/create_env.sh
index 1dd10626d87cb340fc90e7494931f8619a844f8b..fdfe3114e20ad38dc73a92eb5e51f2a1f6974dbe 100755
--- a/video_prediction_tools/env_setup/create_env.sh
+++ b/video_prediction_tools/env_setup/create_env.sh
@@ -88,7 +88,8 @@ fi
 if [[ "$ENV_EXIST" == 0 ]]; then
   # Activate virtual environment and install additional Python packages.
   echo "Configuring and activating virtual environment on ${HOST_NAME}"
-
+  
+  module purge 
   singularity exec --nv "${TF_CONTAINER}" ./install_venv_container.sh "${ENV_DIR}"
   
   info_str="Virtual environment ${ENV_DIR} has been set up successfully."