onnx: add windows cuda support

koush · May 14, 2024 · 390d1b3 · 390d1b3
1 parent 024e997
commit 390d1b3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/plugins/onnx/package-lock.json b/plugins/onnx/package-lock.json
diff --git a/plugins/onnx/package.json b/plugins/onnx/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.88"
+   "version": "0.1.89"
 }
diff --git a/plugins/onnx/src/ort/__init__.py b/plugins/onnx/src/ort/__init__.py
@@ -92,7 +92,7 @@ def __init__(self, nativeId: str | None = None):
                 if sys.platform == 'darwin':
                     providers.append("CoreMLExecutionProvider")
 
-                if 'linux' in sys.platform and platform.machine() == 'x86_64':
+                if ('linux' in sys.platform or 'win' in sys.platform) and platform.machine() == 'x86_64':
                     deviceId = int(deviceId)
                     providers.append(("CUDAExecutionProvider", { "device_id": deviceId }))