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