diff --git a/requirements.txt b/requirements.txt
index f87cc81..bd5149a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,4 @@ setuptools >= 41
 opencv-python-headless
 ocrd >= 2.22.3
 h5py < 3
-tensorflow-gpu >= 2.4.0 
+tensorflow >= 2.4.0