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
|
||||
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
|
||||
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
|
||||
gymnasium>=1.2.3
|
||||
ipykernel==7.2.0
|
||||
jax[cuda12_local]==0.9.0.1
|
||||
scipy>=1.14.0
|
||||
jax[cuda13]==0.9.0.1
|
||||
torch
|
||||
numpy>=2.0.0
|
||||
protobuf>=5.0.0
|
||||
warp-lang
|
||||
|
|
|
|||
|
|
@ -44,11 +44,6 @@ source vsc-venv --activate \
|
|||
set -euo pipefail
|
||||
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...'
|
||||
CLUSTER_ID="${VSC_INSTITUTE_CLUSTER:-generic}"
|
||||
python -m ipykernel install --user --name="sel3_${CLUSTER_ID}" \
|
||||
|
|
|
|||
|
|
@ -50,11 +50,6 @@ source vsc-venv --activate \
|
|||
set -euo pipefail
|
||||
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..."
|
||||
export MUJOCO_GL=egl
|
||||
|
|
|
|||
Reference in a new issue