jax[cuda12] on linux for GPU; EGL headless render; non-fatal video

2026-06-10 19:25:33 +02:00
parent a6fbde798a
commit 4210b6cb53
3 changed files with 38 additions and 30 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,8 @@ hydra-core
 omegaconf
 mujoco==3.5.0
 mujoco-mjx==3.5.0
-jax==0.9.1
+jax[cuda12]==0.9.1 ; sys_platform == "linux"
+jax==0.9.1 ; sys_platform != "linux"
 skrl[torch]==1.4.3
 clearml
 imageio