From 5043be1af97f76d33118a34162f7230579d4d247 Mon Sep 17 00:00:00 2001 From: Jeffrey Wu Date: Fri, 22 Nov 2024 03:12:42 -0500 Subject: [PATCH] updated reqs --- README.md | 2 +- serl_launcher/requirements.txt | 15 +++++++-------- 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index e0870012..7d9e50aa 100644 --- a/README.md +++ b/README.md @@ -49,7 +49,7 @@ We fixed a major issue in the intervention action frame. See release [v0.1.1](ht - For GPU: ```bash - pip install --upgrade "jax[cuda12]==0.4.28" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html + pip install --upgrade "jax[cuda12_pip]==0.4.35" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ``` - For TPU diff --git a/serl_launcher/requirements.txt b/serl_launcher/requirements.txt index 92ddca92..86f92e95 100644 --- a/serl_launcher/requirements.txt +++ b/serl_launcher/requirements.txt @@ -1,15 +1,14 @@ gym >= 0.26 -numpy==1.24.3 -jax>=0.4.13 -distrax==0.1.2 -flax==0.8.0 +numpy>=1.24.3 +flax>=0.8.0 +distrax>=0.1.2 ml_collections >= 0.1.0 tqdm >= 4.60.0 -chex==0.1.85 -optax==0.1.5 -orbax-checkpoint==0.5.10 +chex>=0.1.85 +optax>=0.1.5 +orbax-checkpoint>=0.5.10 absl-py >= 0.12.0 -scipy <= 1.12.0 +scipy==1.11.4 wandb >= 0.12.14 tensorflow>=2.16.0 tensorflow_probability>=0.24.0