Release v2.5.0 (#2074)

* Update Dockerfile

* Release v2.5.0

* Fix docker GPU parent

* Fix changelog

Dockerfile

@@ -1,25 +1,27 @@
 ARG PARENT_IMAGE
 FROM $PARENT_IMAGE
 ARG PYTORCH_DEPS=cpuonly
-ARG PYTHON_VERSION=3.10
+ARG PYTHON_VERSION=3.12
 ARG MAMBA_DOCKERFILE_ACTIVATE=1  # (otherwise python will not be found)
 
 # Install micromamba env and dependencies
 RUN micromamba install -n base -y python=$PYTHON_VERSION \
     pytorch $PYTORCH_DEPS -c conda-forge -c pytorch -c nvidia && \
     micromamba clean --all --yes
 
-ENV CODE_DIR /home/$MAMBA_USER
+ENV CODE_DIR=/home/$MAMBA_USER
 
 # Copy setup file only to install dependencies
 COPY --chown=$MAMBA_USER:$MAMBA_USER ./setup.py ${CODE_DIR}/stable-baselines3/setup.py
 COPY --chown=$MAMBA_USER:$MAMBA_USER ./stable_baselines3/version.txt ${CODE_DIR}/stable-baselines3/stable_baselines3/version.txt
 
 RUN cd ${CODE_DIR}/stable-baselines3 && \
-    pip install -e .[extra,tests,docs] && \
+    pip install uv && \
+    uv pip install --system -e .[extra,tests,docs] && \
     # Use headless version for docker
-    pip uninstall -y opencv-python && \
-    pip install opencv-python-headless && \
-    pip cache purge
+    uv pip uninstall opencv-python && \
+    uv pip install --system opencv-python-headless && \
+    pip cache purge && \
+    uv cache clean
 
 CMD /bin/bash

docs/misc/changelog.rst

@@ -3,9 +3,12 @@
 Changelog
 ==========
 
-Release 2.5.0a1 (WIP)
+Release 2.5.0 (2025-01-27)
 --------------------------
 
+**New algorithm: SimBa in SBX, NumPy 2.0 support**
+
+
 Breaking Changes:
 ^^^^^^^^^^^^^^^^^
 - Increased minimum required version of PyTorch to 2.3.0
@@ -27,12 +30,15 @@ Bug Fixes:
 
 `SBX`_ (SB3 + Jax)
 ^^^^^^^^^^^^^^^^^^
+- Added SimBa Policy: Simplicity Bias for Scaling Up Parameters in DRL
+- Added support for parameter resets
 
 Deprecations:
 ^^^^^^^^^^^^^
 
 Others:
 ^^^^^^^
+- Updated Dockerfile
 
 Documentation:
 ^^^^^^^^^^^^^^

scripts/build_docker.sh

@@ -1,14 +1,14 @@
 #!/bin/bash
 
-CPU_PARENT=mambaorg/micromamba:1.5-jammy
-GPU_PARENT=mambaorg/micromamba:1.5-jammy-cuda-11.7.1
+CPU_PARENT=mambaorg/micromamba:2.0-ubuntu24.04
+GPU_PARENT=mambaorg/micromamba:2.0-cuda12.5.1-ubuntu24.04
 
 TAG=stablebaselines/stable-baselines3
 VERSION=$(cat ./stable_baselines3/version.txt)
 
 if [[ ${USE_GPU} == "True" ]]; then
   PARENT=${GPU_PARENT}
-  PYTORCH_DEPS="pytorch-cuda=11.7"
+  PYTORCH_DEPS="pytorch-cuda=12.4"
 else
   PARENT=${CPU_PARENT}
   PYTORCH_DEPS="cpuonly"

stable_baselines3/version.txt

@@ -1 +1 @@
-2.5.0a1
+2.5.0