diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt
index 9574c8c42..06282147f 100644
--- a/docker/main/requirements-wheels.txt
+++ b/docker/main/requirements-wheels.txt
@@ -12,7 +12,7 @@ pathvalidate == 3.2.*
 markupsafe == 3.0.*
 python-multipart == 0.0.12
 # Classification Model
-tensorflow == 2.19.*
+tensorflow-cpu == 2.19.*
 # General
 mypy == 1.6.1
 onvif-zeep-async == 3.1.*