diff --git a/setup.py b/setup.py index 4b8af07b..ec46174f 100644 --- a/setup.py +++ b/setup.py @@ -36,7 +36,7 @@ "scikit-learn==1.1.1", "xgboost==1.6.1", "lightgbm==3.3.5", - "onnxruntime >=1.10.1, <1.16.0", + "onnxruntime >=1.10.1", "torch>=1.12.1", "protobuf>=3.17.3,<3.21", "pyyaml>=5.4", diff --git a/src/turnkeyml/run/onnxrt/execute.py b/src/turnkeyml/run/onnxrt/execute.py index 48022e09..ba946a94 100644 --- a/src/turnkeyml/run/onnxrt/execute.py +++ b/src/turnkeyml/run/onnxrt/execute.py @@ -9,9 +9,11 @@ import json from statistics import mean import platform +import onnxruntime import turnkeyml.run.plugin_helpers as plugin_helpers -ORT_VERSION = "1.15.1" +# Use the same ORT version of the base environment in the new conda env +ORT_VERSION = onnxruntime.__version__ BATCHSIZE = 1 diff --git a/src/turnkeyml/version.py b/src/turnkeyml/version.py index 159d48b8..0309ae29 100644 --- a/src/turnkeyml/version.py +++ b/src/turnkeyml/version.py @@ -1 +1 @@ -__version__ = "2.0.1" +__version__ = "2.0.2"