From 99459effce6426f42c2300a0c1e928db4aee6f3a Mon Sep 17 00:00:00 2001
From: Augustin Zidek <augustinzidek@google.com>
Date: Fri, 13 Aug 2021 07:29:05 -0700
Subject: [PATCH] Use tensorflow-cpu in setup.py as well.
PiperOrigin-RevId: 390608262
Change-Id: I4c49aeb2aea362d578b7de43d16bf2e418fdd96d
---
setup.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/setup.py b/setup.py
index 762bd7b..0006a2a 100644
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@ setup(
'ml-collections',
'numpy',
'scipy',
- 'tensorflow',
+ 'tensorflow-cpu',
],
tests_require=['mock'],
classifiers=[
--
GitLab