diff --git a/docs/HPC.md b/docs/HPC.md
index b580edd..e2af138 100644
--- a/docs/HPC.md
+++ b/docs/HPC.md
@@ -27,7 +27,7 @@ module swap cluster/joltik
 qsub -I -l nodes=1:ppn=8:gpus=1
 
 # 3. Navigate to the project directory and run the install script
-cd SEL3-2026-Groep-4
+cd "${PBS_O_WORKDIR}"
 bash scripts/hpc/install.sh
 
 # 4. Exit the interactive session once finished
@@ -39,7 +39,7 @@ exit
 
 ## Interactive Debugging on donphan
 
-### Option A — Interactive shell session
+### Interactive shell session
 
 ```bash
 # Swap to the debug cluster (from any login node)
@@ -55,20 +55,18 @@ export UV_CACHE_DIR="$VSC_SCRATCH/.cache/uv"
 # Change to the project directory and activate environment
 cd "$PBS_O_WORKDIR"
 module load vsc-venv
-set +u
 source vsc-venv --activate \
     --modules env/hpc/modules.txt \
     --requirements env/hpc/requirements.txt
-set -u
 
 # Set headless rendering backend
 export MUJOCO_GL=egl
 
 # Run the smoke test
-python src/train.py --config configs/hpc/smoke_test.yaml
+python src/train.py --env-config-path configs/hpc/smoke_test.yaml
 ```
 
-### Option B — JupyterLab session (HPC web portal)
+### JupyterLab session (HPC web portal)
 
 1. In the web portal go to **Interactive Apps → JupyterLab RHEL9**
 2. Set the following options:
diff --git a/env/hpc/modules.txt b/env/hpc/modules.txt
index a710981..c780513 100644
--- a/env/hpc/modules.txt
+++ b/env/hpc/modules.txt
@@ -1,6 +1,6 @@
 gfbf/2024a
 GCCcore/13.3.0
 Python/3.12.3-GCCcore-13.3.0
-PyTorch/2.6.0-foss-2024a-CUDA-12.6.0
+PyTorch/2.7.1-foss-2024a-CUDA-12.6.0
 FFmpeg/7.0.2-GCCcore-13.3.0
 PyYAML/6.0.2-GCCcore-13.3.0
diff --git a/scripts/hpc/install.sh b/scripts/hpc/install.sh
index 8e2c76d..d896064 100644
--- a/scripts/hpc/install.sh
+++ b/scripts/hpc/install.sh
@@ -16,12 +16,14 @@ mkdir -p "$PIP_CACHE_DIR" "$UV_CACHE_DIR"
 
 module load vsc-venv
 
-# Temporarily disable 'unbound variable' check for vsc-venv compatibility
-set +u
+echo 'Installing venv...'
 source vsc-venv --activate \
     --modules env/hpc/modules.txt \
     --requirements env/hpc/requirements.txt
-set -u
 
+echo 'Installing ipykernel...'
 python -m ipykernel install --user --name="sel3_${VSC_INSTITUTE_CLUSTER}" \
     --display-name "SEL3 (${VSC_INSTITUTE_CLUSTER})"
+
+echo 'Done'
+