Switch to Tensorflow CPU-only. GPU not needed for data pipeline.

PiperOrigin-RevId: 387576219 Change-Id: Ic735f068df2a3e9f53bed836f172c08f79ac231c

Switch to Tensorflow CPU-only. GPU not needed for data pipeline.
PiperOrigin-RevId: 387576219 Change-Id: Ic735f068df2a3e9f53bed836f172c08f79ac231c
b3ed8603 · Tim Green · Copybara-Service · bc5547b6 · b3ed8603
Commit b3ed8603 authored Jul 29, 2021 by Tim Green Committed by Copybara-Service Jul 29, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

requirements.txt requirements.txt +1 -1

No files found.
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,4 @@ jax==0.2.14
 ml-collections==0.1.0
 numpy==1.19.5
 scipy==1.7.0
-tensorflow==2.5.0
+tensorflow-cpu==2.5.0