pin skrl/jax/mujoco/gymnasium versions; custom CUDA base image

2026-06-10 09:05:48 +02:00
parent 56499ebe97
commit a6fbde798a
2 changed files with 7 additions and 7 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,11 +1,11 @@
 torch
-gymnasium
+gymnasium==1.2.3
 hydra-core
 omegaconf
-mujoco
-mujoco-mjx
-jax
-skrl[torch]
+mujoco==3.5.0
+mujoco-mjx==3.5.0
+jax==0.9.1
+skrl[torch]==1.4.3
 clearml
 imageio
 imageio-ffmpeg