fix: onnxruntime-gpu is x86_64-only; use onnxruntime on arm64

onnxruntime-gpu has no arm64 wheels (manylinux_2_27_x86_64 /
manylinux_2_28_x86_64 only). uv sync --frozen failed on the arm64
image with exit code 2. Gated onnxruntime-gpu behind the x86_64
marker; arm64 and non-Linux use the CPU onnxruntime package. Added
required-environments so the lockfile is solved for both platforms.
Removed onnxruntime-gpu from override-dependencies (it had no marker
support and blocked arm64 resolution).

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-06-12 05:05:25 +00:00
co-authored by Claude Sonnet 4.6
parent 21208f8331
commit 4cac3d28d2
3 changed files with 63 additions and 45 deletions
+9 -3
View File
@@ -1,6 +1,6 @@
[project]
name = "winnow"
version = "0.2.8"
version = "0.2.9"
description = "Immich to Frigate training sets"
license = "AGPL-3.0-or-later"
requires-python = ">=3.13"
@@ -18,7 +18,9 @@ dependencies = [
"insightface>=0.7.3",
"nvidia-cudnn-cu12>=9.0.0",
"numpy>=2.2.6",
"onnxruntime-gpu>=1.23.2",
"onnxruntime-gpu>=1.23.2; sys_platform == 'linux' and platform_machine == 'x86_64'",
"onnxruntime>=1.23.2; sys_platform == 'linux' and platform_machine != 'x86_64'",
"onnxruntime>=1.23.2; sys_platform != 'linux'",
"opencv-python-headless>=4.12.0.88",
"pillow>=12.1.0",
"python-dotenv>=1.2.1",
@@ -37,7 +39,11 @@ winnow = "winnow.cli:main"
Repository = "https://github.com/sudolulo/winnow"
[tool.uv]
override-dependencies = ["onnxruntime-gpu>=1.23.2"]
override-dependencies = ["onnxruntime>=1.23.2"]
required-environments = [
"sys_platform == 'linux' and platform_machine == 'x86_64'",
"sys_platform == 'linux' and platform_machine == 'aarch64'",
]
[tool.uv.sources]
torch = [