1
Fork 0
This repository has been archived on 2026-08-15. You can view files and clone it, but you cannot make any changes to it's state, such as pushing and creating new issues, pull requests or comments.
2026SEL3-project-Brittle_St.../docs/HPC.md
Tibo De Peuter 92db8c2591 feat(hpc): support run_dir staging and update docs
- 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
2026-04-04 18:59:05 +02:00

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_SCRATCH during the job (fast I/O) and copied to $VSC_DATA at the end for persistence.
  • $VSC_SCRATCH may 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)

  1. In the web portal go to Interactive Apps → JupyterLab RHEL9

  2. 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.0
    Custom code (leave blank — vsc-venv handles modules)
  3. Click Launch, wait for the session to start, then Connect.

  4. In JupyterLab, select the kernel SEL3 (<cluster>).

  5. 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_dir argument. 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.