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

@@ -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

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+.

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 <https://stackoverflow.com/a/30539963>`_. 
+        Some shells such as Zsh require quotation marks around brackets, i.e. ``pip install 'stable-baselines3[extra]'`` `More information <https://stackoverflow.com/a/30539963>`_.
 
 
 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:

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)
 ---------------------------
 

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

stable_baselines3/version.txt

@@ -1 +1 @@
-1.1.0
+1.2.0a0