From 2fa06ae8d24662a40a7c247dd96625bebf182dce Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Tue, 6 Jul 2021 09:32:03 +0200 Subject: [PATCH] Add Python3.9 CI + upgrade min PyTorch version (#503) * Add Python3.9 CI + upgrade min PyTorch version * Upgrade min PyTorch version --- .github/workflows/ci.yml | 4 ++-- README.md | 2 +- docs/guide/install.rst | 4 ++-- docs/misc/changelog.rst | 24 ++++++++++++++++++++++++ setup.py | 2 +- stable_baselines3/version.txt | 2 +- 6 files changed, 31 insertions(+), 7 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index b8579fc..6626122 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -16,7 +16,7 @@ jobs: runs-on: ubuntu-latest strategy: matrix: - python-version: [3.6, 3.7, 3.8] + python-version: [3.6, 3.7, 3.8, 3.9] steps: - uses: actions/checkout@v2 @@ -28,7 +28,7 @@ jobs: run: | python -m pip install --upgrade pip # cpu version of pytorch - pip install torch==1.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html + pip install torch==1.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html pip install .[extra,tests,docs] # Use headless version pip install opencv-python-headless diff --git a/README.md b/README.md index 836e10e..df16898 100644 --- a/README.md +++ b/README.md @@ -79,7 +79,7 @@ Documentation is available online: [https://sb3-contrib.readthedocs.io/](https:/ ## Installation -**Note:** Stable-Baselines3 supports PyTorch 1.4+. +**Note:** Stable-Baselines3 supports PyTorch >= 1.8.1. ### Prerequisites Stable Baselines3 requires python 3.6+. diff --git a/docs/guide/install.rst b/docs/guide/install.rst index 926836e..f43489b 100644 --- a/docs/guide/install.rst +++ b/docs/guide/install.rst @@ -6,7 +6,7 @@ Installation Prerequisites ------------- -Stable-Baselines3 requires python 3.6+. +Stable-Baselines3 requires python 3.6+ and PyTorch >= 1.8.1. Windows 10 ~~~~~~~~~~ @@ -30,7 +30,7 @@ To install Stable Baselines3 with pip, execute: pip install stable-baselines3[extra] .. note:: - Some shells such as Zsh require quotation marks around brackets, i.e. ``pip install 'stable-baselines3[extra]'`` `More information `_. + Some shells such as Zsh require quotation marks around brackets, i.e. ``pip install 'stable-baselines3[extra]'`` `More information `_. This includes an optional dependencies like Tensorboard, OpenCV or ``atari-py`` to train on atari games. If you do not need those, you can use: diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index 5f52552..e3a5daf 100644 --- a/docs/misc/changelog.rst +++ b/docs/misc/changelog.rst @@ -4,6 +4,30 @@ Changelog ========== +Release 1.2.0a0 (WIP) +--------------------------- + +Breaking Changes: +^^^^^^^^^^^^^^^^^ +- SB3 now requires PyTorch >= 1.8.1 + +New Features: +^^^^^^^^^^^^^ + +Bug Fixes: +^^^^^^^^^^ + +Deprecations: +^^^^^^^^^^^^^ + +Others: +^^^^^^^ +- Enabled Python 3.9 in GitHub CI + +Documentation: +^^^^^^^^^^^^^^ + + Release 1.1.0 (2021-07-01) --------------------------- diff --git a/setup.py b/setup.py index a68aa18..4c2b9e9 100644 --- a/setup.py +++ b/setup.py @@ -75,7 +75,7 @@ setup( install_requires=[ "gym>=0.17", "numpy", - "torch>=1.4.0", + "torch>=1.8.1", # For saving models "cloudpickle", # For reading logs diff --git a/stable_baselines3/version.txt b/stable_baselines3/version.txt index 9084fa2..0816bf0 100644 --- a/stable_baselines3/version.txt +++ b/stable_baselines3/version.txt @@ -1 +1 @@ -1.1.0 +1.2.0a0