Bump dependencies

nico-bohlinger · Oct 6, 2024 · 04079ab · 04079ab
1 parent 088a6a1
commit 04079ab
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -71,8 +71,8 @@ git clone [email protected]:nico-bohlinger/RL-X.git
 cd RL-X
 pip install -e .[all] --config-settings editable_mode=compat
 pip uninstall $(pip freeze | grep -i '\-cu12' | cut -d '=' -f 1) -y
-pip install "torch>=2.4.0" --index-url https://download.pytorch.org/whl/cu118 --upgrade
-pip install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+pip install "torch>=2.4.1" --index-url https://download.pytorch.org/whl/cu118 --upgrade
+pip install -U "jax[cuda12]"
 ```
 For other configurations, see the [detailed installation guide](https://nico-bohlinger.github.io/RL-X/#detailed-installation-guide) in the documentation.
 

diff --git a/docs/README.md b/docs/README.md
@@ -78,13 +78,13 @@ pip uninstall $(pip freeze | grep -i '\-cu12' | cut -d '=' -f 1) -y
 ```
 Afterwards, PyTorch can be installed with the following command:
 ```
-pip install "torch>=2.4.0" --index-url https://download.pytorch.org/whl/cu118 --upgrade
+pip install "torch>=2.4.1" --index-url https://download.pytorch.org/whl/cu118 --upgrade
 ```
 
 ### 5. JAX
 For Linux, JAX with GPU support can be installed with the following command:
 ```
-pip install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+pip install -U "jax[cuda12]"
 ```
 For MacOS and Windows, JAX with GPU support is not supported out-of-the-box. However, it can be done with some extra effort (see [here](https://github.com/google/jax) for more information).
 

diff --git a/experiments/colab_experiment.ipynb b/experiments/colab_experiment.ipynb
@@ -15,7 +15,7 @@
    "source": [
     "!git clone https://github.com/nico-bohlinger/RL-X.git && cd RL-X && pip install -e .[all]\n",
     "!pip uninstall $(pip freeze | grep -i '\\-cu12' | cut -d '=' -f 1) -y\n",
-    "!pip install \"torch>=2.4.0\" --index-url https://download.pytorch.org/whl/cu118 --upgrade\n",
+    "!pip install \"torch>=2.4.1\" --index-url https://download.pytorch.org/whl/cu118 --upgrade\n",
     "!pip install -U \"jax[cuda12_pip]\" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"
    ]
   },

diff --git a/experiments/docker/Dockerfile b/experiments/docker/Dockerfile
@@ -8,9 +8,9 @@ RUN git clone https://github.com/nico-bohlinger/RL-X.git
 WORKDIR /RL-X_ws/RL-X
 RUN python3 -m pip install pip -U
 RUN pip3 install -e .[all]
-RUN pip3 install "torch>=2.4.0" --index-url https://download.pytorch.org/whl/cu118 --upgrade
+RUN pip3 install "torch>=2.4.1" --index-url https://download.pytorch.org/whl/cu118 --upgrade
 RUN pip3 uninstall $(pip freeze | grep -i '\-cu12' | cut -d '=' -f 1) -y
-RUN pip3 install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+RUN pip3 install -U "jax[cuda12]"
 
 COPY entrypoint.sh /entrypoint.sh
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -12,18 +12,18 @@ readme = "README.md"
 dependencies = [
     "absl-py>=2.1.0",
     "ml-collections>=0.1.1",
-    "wandb>=0.17.8",
+    "wandb>=0.18.3",
     "gymnasium[mujoco,classic-control,atari,accept-rom-license,other]>=0.29.1",
-    "mujoco>=3.2.2",
-    "mujoco-mjx>=3.2.2",
+    "mujoco>=3.2.3",
+    "mujoco-mjx>=3.2.3",
     "numpy<=1.24.3",
     "tensorflow<=2.15.1",
     "tensorboard<=2.15.2",
-    "jax[cpu]>=0.4.31",
+    "jax[cpu]>=0.4.34",
     "flax>=0.9.0",
-    "orbax_checkpoint>=0.6.1",
+    "orbax_checkpoint>=0.6.4",
     "optax>=0.2.3",
-    "chex>=0.1.86",
+    "chex>=0.1.87",
     "tensorflow_probability>=0.24.0"
 ]