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