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
@@ -106,6 +106,17 @@ setup(
        'orisys': ['*.so', 'configs/*.json'],
    }},
    include_package_data=True,
+    install_requires=[
+        "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"],
+        "cuda": ["cupy-cuda12x>=12.3.0"],  # For backward compatibility, defaults to CUDA 12.x
+    }},
    distclass=BinaryDistribution,
    zip_safe=False,
 )