-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Description
Describe the issue
when I tried to use onnxruntime-cann 1.18.0 and 1.15.0 to run onnx model(sdmgr), it reported the following error:
self.extractor_sdmgr.run(rpt)
result = self.models.get(rtype).run(None, input)
---myenv/lib/python3.8/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py, line 217, in run
return self._sess.run(output_names, input_feed, run_options)
onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running torch-jit-export_6141346329611778433_0 node. Name:'CANNExecutionProvider_torch-jit-export_6141346329611778433_0_0' Status Message: CANN Graph error executing ge::aclgrphBuildModel(graph, options, model)
To reproduce
self.models[type] = ort.InferenceSession(self.checkpoints.get(type), providers=['CANNExecutionProvider'])
result = self.models.get(rtype).run(None, input)
Urgency
No response
Platform
Linux
OS Version
EulerOS 2.0 (SP10)
ONNX Runtime Installation
Released Package
ONNX Runtime Version or Commit ID
onnxruntime_cann-1.15.0/1.18.0
ONNX Runtime API
Python
Architecture
ARM64
Execution Provider
Other / Unknown
Execution Provider Library Version
CANN-8.0