From dff87bc2c1f8ca1d7b7b07dc4699d7af813b6a7e Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 29 May 2025 15:32:45 -0600 Subject: [PATCH] Use tensorflow cpu on amd64 --- docker/main/requirements-wheels.txt | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt index 06282147f..624983eb4 100644 --- a/docker/main/requirements-wheels.txt +++ b/docker/main/requirements-wheels.txt @@ -11,8 +11,9 @@ joserfc == 1.0.* pathvalidate == 3.2.* markupsafe == 3.0.* python-multipart == 0.0.12 -# Classification Model -tensorflow-cpu == 2.19.* +# Classification Model Training +tensorflow == 2.19.* ; platform_machine == 'aarch64' +tensorflow-cpu == 2.19.* ; platform_machine == 'x86_64' # General mypy == 1.6.1 onvif-zeep-async == 3.1.*