- Add run_dir and checkpoint_frequency to PPOArgs - Update train.py to use run_dir for SummaryBoard, model saving, and loss plots - Create configs/production_training.yaml for HPC production runs - Update HPC.md with run_dir staging strategy details
4 KiB
HPC Guide
Full documentation: https://docs.hpc.ugent.be/
Cluster Selection
Choose the appropriate cluster before submitting a job with module swap cluster/<name>. The default login cluster is doduo.
Check the current queue load at https://shieldon.ugent.be:8083/pbsmon-web-users/.
Storage Overview
- Run outputs are written to
$VSC_SCRATCHduring the job (fast I/O) and copied to$VSC_DATAat the end for persistence. $VSC_SCRATCHmay be purged periodically — do not use it as long-term storage.
Check your quota: https://account.vscentrum.be (Usage section).
Initial Environment Setup
Run once from a login shell on donphan after cloning the repository:
# 1. Connect via the web portal → HPC Login → Interactive Apps > Shell (tmux)
# Set cluster to "donphan (interactive/debug)"
# 2. Clone the project into $VSC_HOME (if not done yet)
cd $VSC_HOME
git clone <repo-url>
cd 2026SEL3-project-BrittleStar
# 3. Run the install script
bash scripts/hpc/install.sh
Note: virtual environments are cluster-specific. Re-run the script when switching to a new cluster.
Interactive Debugging on donphan
Option A — Interactive shell session
# Swap to the debug cluster (from any login node)
module swap cluster/donphan
# Request an interactive job (1 node, 4 cores)
qsub -I -l nodes=1:ppn=4 -l walltime=1:00:00
# Once inside the job — redirect caches to scratch first
export PIP_CACHE_DIR="$VSC_SCRATCH/.cache/pip"
export UV_CACHE_DIR="$VSC_SCRATCH/.cache/uv"
# Change to the project directory and activate environment
cd "$PBS_O_WORKDIR"
module load vsc-venv
source vsc-venv --activate --modules env/hpc/modules.txt
# Set headless rendering backend
export MUJOCO_GL=egl
# Run the smoke test
python src/train.py --config configs/hpc/smoke_test.yaml
Option B — JupyterLab session (HPC web portal)
-
In the web portal go to Interactive Apps → JupyterLab RHEL9
-
Set the following options:
Option Value Cluster donphan (interactive/debug)Number of nodes 1 Number of cores 4 JupyterLab version 4.2.5 GCCcore-13.3.0Custom code (leave blank — vsc-venv handles modules) -
Click Launch, wait for the session to start, then Connect.
-
In JupyterLab, select the kernel
SEL3 (<cluster>). -
Verify GPU access:
import jax print(jax.default_backend()) # expected: 'gpu' print(jax.devices()) # expected: [CudaDevice(id=0)]
Warning: JAX can only be loaded by one kernel at a time. Shut down other kernels before switching notebooks.
Submitting Batch Training Jobs
# Default cluster (joltik — one A100 GPU slice)
qsub scripts/hpc/train.pbs
# Switch to a different GPU cluster first
module swap cluster/accelgor
qsub scripts/hpc/train.pbs
The job script automatically:
- Writes run outputs to
$VSC_SCRATCH/runs/<job_id>during the run using the--run_dirargument. This ensures that frequent I/O (like tensorboard logs and checkpoints) happens on the fastest available filesystem. - Copies the final results to
$VSC_DATA/runs/<job_id>on completion for long-term persistence. - Writes PBS stdout/stderr to
runs/brittlestar-ppo.o<job_id>/.e<job_id>(standard PBS convention, relative to the project root).
Monitor your jobs:
qstat # list your jobs
qstat -f <id> # detailed info for a specific job
qdel <id> # cancel a job
Managing Dependencies
env/hpc/requirements.txt is auto-generated by CI whenever pyproject.toml changes. To regenerate locally:
uv run scripts/export_hpc_requirements.py
Do not edit env/hpc/requirements.txt by hand — edit pyproject.toml instead.