Merge pull request #20 from bertsky/patch-1

require tensorflow-gpu instead of tensorflow
qurator-spk · Nov 23, 2020 · 5ef0ee8 · 5ef0ee8
2 parents 8c74d71 + a046699
commit 5ef0ee8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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