Merge pull request #20 from bertsky/patch-1

require tensorflow-gpu instead of tensorflow
4 years ago · 5ef0ee81c8
parent 8c74d717fc a046699b0b
commit 5ef0ee81c8
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