正常安装的amct_onnx在使用时异常
收藏回复举报
正常安装的amct_onnx在使用时异常
t('forum.solved') 已解决
发表于2023-07-07 11:13:34
0 查看

我的是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'

本帖最后由 匿名用户2023/07/07 11:13:36 编辑

我要发帖子