Add upper-bound for Tensorflow version (<2.0.0) (#499)

* Add upper-bound for tensorflow version (<2.0.0) * Update docs on required tensorflow versions
Stable-Baselines-Team · Oct 6, 2019 · feeaff4 · feeaff4
1 parent 0017e62
commit feeaff4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -65,6 +65,8 @@ Documentation: https://stable-baselines.readthedocs.io/en/master/guide/rl_zoo.ht
 
 ## Installation
 
+**Note:** Stabe-Baselines supports Tensorflow versions from 1.8.0 to 1.14.0. Support for Tensorflow 2 API is planned.
+
 ### Prerequisites
 Baselines requires python3 (>=3.5) with the development headers. You'll also need system packages CMake, OpenMPI and zlib. Those can be installed as follows
 

diff --git a/docs/guide/install.rst b/docs/guide/install.rst
@@ -10,6 +10,12 @@ Baselines requires python3 (>=3.5) with the development headers. You'll
 also need system packages CMake, OpenMPI and zlib. Those can be
 installed as follows
 
+.. note::
+
+	Stable-Baselines supports Tensorflow versions from 1.8.0 to 1.14.0, and does not work on
+	Tensorflow versions 2.0.0 and above. Support for Tensorflow 2 API is planned.
+
+
 Ubuntu
 ~~~~~~
 

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 
 tf_dependency = []
 if install_tf:
-    tf_dependency = ['tensorflow-gpu>=1.8.0'] if tf_gpu else ['tensorflow>=1.8.0']
+    tf_dependency = ['tensorflow-gpu>=1.8.0,<2.0.0'] if tf_gpu else ['tensorflow>=1.8.0,<2.0.0']
     if tf_gpu:
         print("A GPU was detected, tensorflow-gpu will be installed")