1
Fork 0

fix(hpc): add PyTorch module to minimize uv

This commit is contained in:
Tibo De Peuter 2026-04-04 20:50:07 +02:00
parent 2d08023abb
commit 745c09e524

1
env/hpc/modules.txt vendored
View file

@ -1,3 +1,4 @@
PyTorch/2.1.2-foss-2023a-CUDA-12.1.1
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