diff --git a/src/python/setup.py b/src/python/setup.py index 8d0a115885..587c2d643b 100644 --- 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",