diff --git a/plugins/devices/src/turnkeyml_plugin_devices/onnxrt/execute.py b/plugins/devices/src/turnkeyml_plugin_devices/onnxrt/execute.py index 794adb11..28f7283f 100644 --- 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 index 5933accc..fad34693 100644 --- 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",