1
Fork 0

feat(hpc): add structured env/hpc config and requirements export script

- env/hpc/modules.txt: curated list of HPC EasyBuild modules loaded by vsc-venv
- env/hpc/requirements.txt: pip packages not covered by HPC modules (auto-generated)
- scripts/export_hpc_requirements.py: developer utility to regenerate requirements.txt
  from pyproject.toml; to be called by CI when pyproject.toml changes
This commit is contained in:
Tibo De Peuter 2026-04-04 18:10:38 +02:00
parent f2139fb419
commit 23f0b45614
3 changed files with 92 additions and 0 deletions

7
env/hpc/modules.txt vendored Normal file
View file

@ -0,0 +1,7 @@
jax/0.4.25-gfbf-2023a-CUDA-12.1.1
Flax/0.8.4-gfbf-2023a-CUDA-12.1.1
Optax/0.2.2-gfbf-2023a-CUDA-12.1.1
wandb/0.16.1-GCC-12.3.0
matplotlib/3.7.2-gfbf-2023a
PyYAML/6.0-GCCcore-12.3.0
FFmpeg/5.1.2-GCCcore-12.3.0

9
env/hpc/requirements.txt vendored Normal file
View file

@ -0,0 +1,9 @@
biorobot==0.4.2
cleanrl>=0.4.8
evosax==0.2.0
gymnasium>=1.2.3
ipykernel==7.2.0
mediapy==1.2.6
pyopengl>=3.1.10
pyopengl-accelerate>=3.1.10
tyro>=1.0.10