From a3e55582f77f28eb8641661de7289b9bfe548b0f Mon Sep 17 00:00:00 2001 From: Clemens Neudecker <952378+cneud@users.noreply.github.com> Date: Fri, 4 Dec 2020 21:16:58 +0100 Subject: [PATCH] tensorflow-gpu instead of tensorflow --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index dcef941..285eead 100644 --- 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