pytorch转onnx模型后,对onnx模型进行runtime时提示以下错误,具体细节如下:
onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running FusedConv node. Name:'Conv_461_Add_463_Relu_464' Status Message: CUDNN error executing cudnnSetTensorNdDescriptor(tensor_, dataType, static_cast(rank), dims.data(), strides.data())。
pdf文件中是对该错误的解决思路。
2022-05-16 21:05:45
173KB
文档资料