Restrict TensorFlow Version on Linux (#3089)

apple · Apr 3, 2020 · 3473ac2 · 3473ac2
1 parent 73dad21
commit 3473ac2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/python/setup.py b/src/python/setup.py
@@ -135,8 +135,13 @@ def run(self):
         "requests >= 2.9.1",
         "scipy >= 1.1.0",
         "six >= 1.10.0",
-        "tensorflow >= 2.0.0",
     ]
+    if sys.platform == "darwin":
+        install_requires.append("tensorflow >= 2.0.0")
+    else:
+        # ST, OD, AC and DC segfault on Linux with TensorFlow 2.1
+        # See: https://github.com/apple/turicreate/issues/3003
+        install_requires.append("tensorflow >= 2.0.0,<= 2.0.1 ")
 
     setup(
         name="turicreate",