fix(hpc): add PyTorch module to minimize uv
This commit is contained in:
parent
2d08023abb
commit
745c09e524
1 changed files with 1 additions and 0 deletions
1
env/hpc/modules.txt
vendored
1
env/hpc/modules.txt
vendored
|
|
@ -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
|
||||
|
|
|
|||
Reference in a new issue