From 04079ab6f83d9dccffef97f63a2d4c4466c6e64f Mon Sep 17 00:00:00 2001 From: Nico Date: Sun, 6 Oct 2024 13:57:09 +0200 Subject: [PATCH] Bump dependencies --- README.md | 4 ++-- docs/README.md | 4 ++-- experiments/colab_experiment.ipynb | 2 +- experiments/docker/Dockerfile | 4 ++-- pyproject.toml | 12 ++++++------ 5 files changed, 13 insertions(+), 13 deletions(-) diff --git a/README.md b/README.md index e5fd056..bb9a24c 100644 --- a/README.md +++ b/README.md @@ -71,8 +71,8 @@ git clone git@github.com: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 index f311f24..2212460 100644 --- 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 index 63b3059..eed332c 100644 --- 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 index 0687050..ae334ff 100644 --- 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 index 3f75861..34264c1 100644 --- 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" ]