onnxruntime可以导出模型,但使用onnx推理时程序异常退出如下:
Process finished with exit code 139
推理时使用了以下模板,执行到ort_session.run后就会异常错误码退出。
import onnxruntime
onnx_file_name = "xxxxxx.onnx"
ort_session = onnxruntime.InferenceSession(onnx_file_name)
ort_inputs = {'input': input_img}
ort_output = ort_session.run(None,ort_inputs)[0]
根据官网依赖搭配列表检查发现,是onnxruntime-gpu的版本安装太高了。本机器的CUDA11.7适合搭配onnxruntime-gpu 1.15到1.17,但我安装了1.17.x。降级到1.15后就好了。