Unverified Commit 0efdf411 authored by xieofxie's avatar xieofxie Committed by GitHub
Browse files

add onnxruntime-qnn & onnxruntime-cann (#11269)


Co-authored-by: default avatarhualxie <hualxie@microsoft.com>
parent 450dc48a
......@@ -101,18 +101,20 @@ _onnx_available = importlib.util.find_spec("onnxruntime") is not None
if _onnx_available:
candidates = (
"onnxruntime",
"onnxruntime-cann",
"onnxruntime-directml",
"ort_nightly_directml",
"onnxruntime-gpu",
"ort_nightly_gpu",
"onnxruntime-directml",
"onnxruntime-migraphx",
"onnxruntime-openvino",
"ort_nightly_directml",
"onnxruntime-qnn",
"onnxruntime-rocm",
"onnxruntime-migraphx",
"onnxruntime-training",
"onnxruntime-vitisai",
)
_onnxruntime_version = None
# For the metadata, we have to look for both onnxruntime and onnxruntime-gpu
# For the metadata, we have to look for both onnxruntime and onnxruntime-x
for pkg in candidates:
try:
_onnxruntime_version = importlib_metadata.version(pkg)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment