本文章主要探讨和解决在使用ONNX Runtime过程中遇到的一个常见错误——[ONNXRuntimeError]:1 :FAIL非零状态码返回的问题,分析其原因并提供解决方案。
在使用PyTorch模型并将其转换为ONNX格式后,在对生成的ONNX模型进行runtime操作时遇到了错误提示:onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : 在执行FusedConv节点时返回了非零状态码。具体来说,该错误与名为Conv_461_Add_463_Relu_464的节点有关,并且提示信息为CUDNN在设置张量描述符时遇到错误:cudnnSetTensorNdDescriptor(tensor_, dataType, static_cast(rank), dims.data(), strides.data())。此外,有一个PDF文件提供了该问题的具体解决思路。