深度学习模型推理onnxruntimer报错
深度学习模型推理onnxruntimer报错
·
onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Load model from liveportrait/warping_spade.onnx failed:Node (/dense_motion_network/GridSample) Op (GridSample) [ShapeInferenceError] Input 0 expected to have rank 4 but has rank 5
发生这个原因是因为ONNX模型文件不兼容ONNX Runtime当前版本所导致的,可以查看当前onnxrumtime的版本:
pip list
卸掉当前onnxruntime版本:
pip uninstall onnxruntime
查找当前能安装的所有onnxruntime版本:
pip install onnxruntime==
安装最新的onnxruntime最新版本:
pip install onnxruntime==1.19.2
更多推荐
已为社区贡献6条内容
所有评论(0)