tensorflow-gpu instead of tensorflow

4 years ago · a3e55582f7
parent 30829bc9b2
commit a3e55582f7
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -2,5 +2,5 @@
 ocrd >= 2.20.1
 keras >= 2.3.1
 scikit-learn >= 0.23.2
-tensorflow >= 1.15, < 2
+tensorflow-gpu >= 1.15, < 2
 imutils >= 0.5.3