tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

fix(hpc): cuda

Browse source
This commit is contained in:
Tibo De Peuter 2026-04-04 23:35:48 +02:00
parent 7e902aa8b6
commit 6065603fbd
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
env/hpc/requirements.txt vendored
View file

@ -4,7 +4,7 @@ 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==0.9.0.1 jax[cuda12]==0.9.0.1
numpy>=2.0.0 numpy>=2.0.0
scipy>=1.14.0 scipy>=1.14.0
protobuf>=5.0.0 protobuf>=5.0.0

2
pyproject.toml
View file

@ -28,7 +28,7 @@ dependencies = [
[project.optional-dependencies] [project.optional-dependencies]
cuda = [ cuda = [
"jax[cuda13]==0.9.0.1", "jax[cuda12]==0.9.0.1",
] ]
[dependency-groups] [dependency-groups]