diff --git a/Dockerfile b/Dockerfile index 1219bceb1..040f91ba7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ 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 @@ -9,17 +9,19 @@ 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 diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index a9eef9447..cf0db00fa 100644 --- a/docs/misc/changelog.rst +++ b/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: ^^^^^^^^^^^^^^ diff --git a/scripts/build_docker.sh b/scripts/build_docker.sh index 10c8689a0..951c66983 100755 --- a/scripts/build_docker.sh +++ b/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" diff --git a/stable_baselines3/version.txt b/stable_baselines3/version.txt index 240183827..437459cd9 100644 --- a/stable_baselines3/version.txt +++ b/stable_baselines3/version.txt @@ -1 +1 @@ -2.5.0a1 +2.5.0