Use tensorflow-cpu in setup.py as well.

PiperOrigin-RevId: 390608262 Change-Id: I4c49aeb2aea362d578b7de43d16bf2e418fdd96d

Use tensorflow-cpu in setup.py as well.
PiperOrigin-RevId: 390608262 Change-Id: I4c49aeb2aea362d578b7de43d16bf2e418fdd96d
99459eff · Augustin Zidek · Copybara-Service · cef198e0 · 99459eff
Commit 99459eff authored Aug 13, 2021 by Augustin Zidek Committed by Copybara-Service Aug 13, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@ setup(
        'ml-collections',
        'numpy',
        'scipy',
-        'tensorflow',
+        'tensorflow-cpu',
    ],
    tests_require=['mock'],
    classifiers=[