fixup! fixup! fix(hpc): rework modules again
This commit is contained in:
parent
3e586da953
commit
059ac681ec
4 changed files with 2 additions and 15 deletions
3
env/hpc/modules.txt
vendored
3
env/hpc/modules.txt
vendored
|
|
@ -1,6 +1,3 @@
|
||||||
gfbf/2024a
|
|
||||||
GCCcore/13.3.0
|
GCCcore/13.3.0
|
||||||
Python/3.12.3-GCCcore-13.3.0
|
Python/3.12.3-GCCcore-13.3.0
|
||||||
PyTorch/2.7.1-foss-2024a-CUDA-12.6.0
|
|
||||||
FFmpeg/7.0.2-GCCcore-13.3.0
|
FFmpeg/7.0.2-GCCcore-13.3.0
|
||||||
PyYAML/6.0.2-GCCcore-13.3.0
|
|
||||||
|
|
|
||||||
4
env/hpc/requirements.txt
vendored
4
env/hpc/requirements.txt
vendored
|
|
@ -4,8 +4,8 @@ evosax==0.2.0
|
||||||
flax>=0.12.2
|
flax>=0.12.2
|
||||||
gymnasium>=1.2.3
|
gymnasium>=1.2.3
|
||||||
ipykernel==7.2.0
|
ipykernel==7.2.0
|
||||||
jax[cuda12_local]==0.9.0.1
|
jax[cuda13]==0.9.0.1
|
||||||
scipy>=1.14.0
|
torch
|
||||||
numpy>=2.0.0
|
numpy>=2.0.0
|
||||||
protobuf>=5.0.0
|
protobuf>=5.0.0
|
||||||
warp-lang
|
warp-lang
|
||||||
|
|
|
||||||
|
|
@ -44,11 +44,6 @@ source vsc-venv --activate \
|
||||||
set -euo pipefail
|
set -euo pipefail
|
||||||
cd "$PBS_O_WORKDIR"
|
cd "$PBS_O_WORKDIR"
|
||||||
|
|
||||||
# Prepend venv to PYTHONPATH to shadow system packages (e.g. SciPy-bundle compiled
|
|
||||||
# against NumPy 1.x) that are injected by the loaded modules.
|
|
||||||
VENV_LIB_DIR="$VIRTUAL_ENV/lib/python$(python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')/site-packages"
|
|
||||||
export PYTHONPATH="$VENV_LIB_DIR:${PYTHONPATH:-}"
|
|
||||||
|
|
||||||
echo '>>> Installing ipykernel...'
|
echo '>>> Installing ipykernel...'
|
||||||
CLUSTER_ID="${VSC_INSTITUTE_CLUSTER:-generic}"
|
CLUSTER_ID="${VSC_INSTITUTE_CLUSTER:-generic}"
|
||||||
python -m ipykernel install --user --name="sel3_${CLUSTER_ID}" \
|
python -m ipykernel install --user --name="sel3_${CLUSTER_ID}" \
|
||||||
|
|
|
||||||
|
|
@ -50,11 +50,6 @@ source vsc-venv --activate \
|
||||||
set -euo pipefail
|
set -euo pipefail
|
||||||
cd "$PBS_O_WORKDIR"
|
cd "$PBS_O_WORKDIR"
|
||||||
|
|
||||||
# Prepend venv to PYTHONPATH to shadow system packages (e.g. SciPy-bundle compiled
|
|
||||||
# against NumPy 1.x) that are injected by the loaded modules.
|
|
||||||
VENV_LIB_DIR="$VIRTUAL_ENV/lib/python$(python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')/site-packages"
|
|
||||||
export PYTHONPATH="$VENV_LIB_DIR:${PYTHONPATH:-}"
|
|
||||||
|
|
||||||
|
|
||||||
echo ">>> Starting BrittleStar training..."
|
echo ">>> Starting BrittleStar training..."
|
||||||
export MUJOCO_GL=egl
|
export MUJOCO_GL=egl
|
||||||
|
|
|
||||||
Reference in a new issue