From a046699b0b270dbd2b3c816b1e53523a5442d669 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky <38561704+bertsky@users.noreply.github.com> Date: Mon, 23 Nov 2020 10:07:58 +0100 Subject: [PATCH] require tensorflow-gpu instead of tensorflow --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 37124ef..6b012f7 100644 --- 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