File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -101,18 +101,20 @@ def _is_package_available(pkg_name: str):
101101if _onnx_available :
102102 candidates = (
103103 "onnxruntime" ,
104+ "onnxruntime-cann" ,
105+ "onnxruntime-directml" ,
106+ "ort_nightly_directml" ,
104107 "onnxruntime-gpu" ,
105108 "ort_nightly_gpu" ,
106- "onnxruntime-directml " ,
109+ "onnxruntime-migraphx " ,
107110 "onnxruntime-openvino" ,
108- "ort_nightly_directml " ,
111+ "onnxruntime-qnn " ,
109112 "onnxruntime-rocm" ,
110- "onnxruntime-migraphx" ,
111113 "onnxruntime-training" ,
112114 "onnxruntime-vitisai" ,
113115 )
114116 _onnxruntime_version = None
115- # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu
117+ # For the metadata, we have to look for both onnxruntime and onnxruntime-x
116118 for pkg in candidates :
117119 try :
118120 _onnxruntime_version = importlib_metadata .version (pkg )
You can’t perform that action at this time.
0 commit comments