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
|
||||
|
|
|
|||
Reference in a new issue