Relax TensorFlow Requirement (#3281)

apple · Aug 5, 2020 · 25ff85d · 25ff85d
1 parent 24c5102
commit 25ff85d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/setup.py b/src/python/setup.py
@@ -182,9 +182,9 @@ def do_not_install(require):
     if sys.platform == "darwin":
         install_requires.append("tensorflow >= 2.0.0")
     else:
-        # ST, OD, AC and DC segfault on Linux with TensorFlow 2.1
+        # ST, OD, AC and DC segfault on Linux with TensorFlow 2.1.0 and 2.1.1
         # See: https://github.com/apple/turicreate/issues/3003
-        install_requires.append("tensorflow >= 2.0.0,<= 2.0.1 ")
+        install_requires.append("tensorflow >= 2.0.0,!= 2.1.0,!= 2.1.1")
 
     setup(
         name="turicreate",