See if ORT-DML dep works just as well as ORT dep

onnx · Nov 18, 2024 · e5198ff · e5198ff
1 parent ff12b67
commit e5198ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/plugins/devices/src/turnkeyml_plugin_devices/onnxrt/execute.py b/plugins/devices/src/turnkeyml_plugin_devices/onnxrt/execute.py
@@ -12,8 +12,8 @@
 import onnxruntime
 import turnkeyml_plugin_devices.common.run.plugin_helpers as plugin_helpers
 
-# Use the same ORT version of the base environment in the new conda env
-ORT_VERSION = onnxruntime.__version__
+# Use the last ORT release to support Python 3.8
+ORT_VERSION = "1.19.2"
 
 BATCHSIZE = 1
 

diff --git a/setup.py b/setup.py
@@ -35,7 +35,7 @@
         #   ImportError: DLL load failed while importing onnx_cpp2py_export
         "onnx>=1.11.0,<1.16.2",
         "onnxmltools==1.10.0",
-        "onnxruntime >=1.10.1",
+        "onnxruntime-directml >=1.10.1",
         "torch>=1.12.1",
         "pyyaml>=5.4",
         "typeguard>=2.3.13",