diff --git a/env/hpc/requirements.txt b/env/hpc/requirements.txt
index 20a1fe1..eeca95a 100644
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@@ -4,7 +4,7 @@ evosax==0.2.0
 flax>=0.12.2
 gymnasium>=1.2.3
 ipykernel==7.2.0
-jax[cuda13]==0.9.0.1
+jax[cuda12_local]==0.9.0.1
 scipy>=1.14.0
 numpy>=2.0.0
 protobuf>=5.0.0