update example and build

2026-01-10 19:20:17 +08:00
parent cd7ab65e84
commit dff2aef2fa
8 changed files with 123 additions and 20 deletions
@@ -19,9 +19,16 @@ setup(
    },
    include_package_data=True,
    install_requires=[
-        "numpy",
-        "opencv-python",
-        "scipy",
+        "numpy>=1.24.4",
+        "opencv-python>=4.10.0.84",
+        "opencv-contrib-python",
+        "scipy>=1.10.1",
    ],
+    extras_require={
+        "cuda12": ["cupy-cuda12x>=12.3.0"],
+        "cuda13": ["cupy-cuda13x>=13.0.0"],
+        # For backward compatibility, 'cuda' defaults to CUDA 12.x
+        "cuda": ["cupy-cuda12x>=12.3.0"],
+    },
    python_requires=">=3.8",
 )