diff --git a/requirements.txt b/requirements.txt
index fc2ed16..94a797e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 numpy
-tensorflow == 1.15.*
+tensorflow-gpu == 1.15.2
 calamari-ocr == 0.3.5
 setuptools >= 41.0.0  # tensorboard depends on this, but why do we get an error at runtime?
 click