fix: force onnxruntime-gpu over CPU + preload CUDA DLLs

2026-06-11 10:34:31 -04:00
parent dac2bc8242
commit a04d1a5d9b
4 changed files with 45 additions and 46 deletions
@@ -18,7 +18,7 @@ dependencies = [
    "insightface>=0.7.3",
    "nvidia-cudnn-cu12>=9.0.0",
    "numpy>=2.2.6",
-    "onnxruntime-gpu==1.20.1",
+    "onnxruntime-gpu>=1.23.2",
    "opencv-python-headless>=4.12.0.88",
    "pillow>=12.1.0",
    "python-dotenv>=1.2.1",
@@ -35,6 +35,9 @@ if-curator = "if_curator.cli:main"
 [project.urls]
 Repository = "https://github.com/sudolulo/if_curator_headless"

+[tool.uv]
+override-dependencies = ["onnxruntime-gpu>=1.23.2"]
+
 [tool.uv.sources]
 torch = [
    { index = "pytorch-cu126", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },