require tensorflow-gpu instead of tensorflow

4 years ago · a046699b0b
parent 8c74d717fc
commit a046699b0b
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -4,4 +4,4 @@ opencv-python-headless
 ocrd >= 2.18.0
 keras >= 2.3.1, < 2.4
 h5py < 3
-tensorflow >= 1.15, < 1.16
+tensorflow-gpu >= 1.15, < 1.16