我的是Atlas 200 的RC模式,系统是Ubuntu 18.04.4
驱动版本是21.03
工具版本是6.3.RC2
异常如下,请问如何消除异常:
HwHiAiUser@davinci-mini:~/amct/amct_onnx/amct_onnx_op$ python3
Python 3.9.2 (default, Jul 6 2023, 08:34:40)
[GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import amct_onnx
2023-07-07 01:35:06,870 - WARNING - [AMCT]:[AMCT]: [ONNXRuntimeError] : 1 : FAIL : Failed to load library /home/HwHiAiUser/.local/lib/python3.9/site-packages/amct_onnx/custom_op/libamct_onnx_ops.so with error: /home/HwHiAiUser/.local/lib/python3.9/site-packages/amct_onnx/custom_op/libamct_onnx_ops.so: undefined symbol: _ZNK4util24DataCastToFloat32FunctorINS_9GpuDeviceEtEclEPKtPfi
在量化自己的模型和官方示例时均报错:
Traceback (most recent call last):
File "/home/HwHiAiUser/code/resnet-101/src/resnet-101_calibration.py", line 149, in <module>
main()
File "/home/HwHiAiUser/code/resnet-101/src/resnet-101_calibration.py", line 134, in main
onnx_forward(modified_model, 1, batch_num)
File "/home/HwHiAiUser/code/resnet-101/src/resnet-101_calibration.py", line 86, in onnx_forward
ort_session = ort.InferenceSession(onnx_model, amct.AMCT_SO)
File "/home/HwHiAiUser/.local/lib/python3.9/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 283, in __init__
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "/home/HwHiAiUser/.local/lib/python3.9/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 310, in _create_inference_session
sess = C.InferenceSession(session_options, self._model_path, True, self._read_config_from_model)
onnxruntime.capi.onnxruntime_pybind11_state.InvalidGraph: [ONNXRuntimeError] : 10 : INVALID_GRAPH : Load model from /home/HwHiAiUser/code/resnet-101/outputs/calibration/tmp/modified_model.onnx failed:This is an invalid model. Error in Node:/model.0/conv/Conv_ifmr_op : No opset import for domain 'amct.customop'
我的是Atlas 200 的RC模式,系统是Ubuntu 18.04.4
驱动版本是21.03
工具版本是6.3.RC2
异常如下,请问如何消除异常:
HwHiAiUser@davinci-mini:~/amct/amct_onnx/amct_onnx_op$ python3
Python 3.9.2 (default, Jul 6 2023, 08:34:40)
[GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import amct_onnx
2023-07-07 01:35:06,870 - WARNING - [AMCT]:[AMCT]: [ONNXRuntimeError] : 1 : FAIL : Failed to load library /home/HwHiAiUser/.local/lib/python3.9/site-packages/amct_onnx/custom_op/libamct_onnx_ops.so with error: /home/HwHiAiUser/.local/lib/python3.9/site-packages/amct_onnx/custom_op/libamct_onnx_ops.so: undefined symbol: _ZNK4util24DataCastToFloat32FunctorINS_9GpuDeviceEtEclEPKtPfi
在量化自己的模型和官方示例时均报错:
Traceback (most recent call last):
File "/home/HwHiAiUser/code/resnet-101/src/resnet-101_calibration.py", line 149, in <module>
main()
File "/home/HwHiAiUser/code/resnet-101/src/resnet-101_calibration.py", line 134, in main
onnx_forward(modified_model, 1, batch_num)
File "/home/HwHiAiUser/code/resnet-101/src/resnet-101_calibration.py", line 86, in onnx_forward
ort_session = ort.InferenceSession(onnx_model, amct.AMCT_SO)
File "/home/HwHiAiUser/.local/lib/python3.9/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 283, in __init__
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "/home/HwHiAiUser/.local/lib/python3.9/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 310, in _create_inference_session
sess = C.InferenceSession(session_options, self._model_path, True, self._read_config_from_model)
onnxruntime.capi.onnxruntime_pybind11_state.InvalidGraph: [ONNXRuntimeError] : 10 : INVALID_GRAPH : Load model from /home/HwHiAiUser/code/resnet-101/outputs/calibration/tmp/modified_model.onnx failed:This is an invalid model. Error in Node:/model.0/conv/Conv_ifmr_op : No opset import for domain 'amct.customop'