![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
onnx
York1996
懒人一枚,热爱提高效率,简化流程,自动办公;
展开
-
导出的onnx的所有节点都没有name属性的可能原因
允许 PyTorch 在导出 ONNX 模型时使用 ATen 操作符作为回退,以确保所有操作符都能被正确导出。这对于模型中包含非标准或自定义操作符的情况尤其有用。但是可能回导致节点没有name属性;原创 2024-08-04 13:07:45 · 338 阅读 · 0 评论 -
onnx.onnx_cpp2py_export.checker.ValidationError: Nodes in a graph must be topologically sorted的可能原因
上述报错的可能原因是,只修改了onnx中每个node的input,output属性和output的属性。可以通过调用下面的函数,看onnx格式是否正确:(如果不能被化简,则说明onnx格式不正确)原创 2023-06-05 16:58:14 · 1923 阅读 · 0 评论 -
onnxruntime安装了GPU版本,但是还是使用cpu的可能原因
pytorch等可以用cuda来运行。确保指定cuda来运行。ort_session=ort.InferenceSession(onnx_path,providers=["CUDAExecutionProvider"]) # CUDAExecutionProvider CPUExecutionProvider可能原因是没有安装cudnn,可以用conda install cudnn试试。gpu版本比cpu版本还是要快四五倍的。...原创 2022-06-30 18:11:12 · 2662 阅读 · 0 评论 -
下面问题的可能原因
onnxruntime.capi.onnxruntime_pybind11_state.RuntimeException: [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : Non-zero status code returned while running ArgMax node. Name:'ArgMax_84' Status Message: /onnxruntime_src/onnxruntime/core/framework/bfc_arena.cc:34原创 2022-06-16 11:55:17 · 4556 阅读 · 4 评论 -
AttributeError: module ‘onnxruntime‘ has no attribute ‘InferenceSession‘的可能解决方法
AttributeError: module 'onnxruntime' has no attribute 'InferenceSession'可能原因是需要卸载之后重新安装onnxruntime。pip uninstallpip install原创 2022-06-16 11:36:38 · 2739 阅读 · 0 评论