博客讨论了在使用jetson-inference库将onnx模型转换为TensorRT时遇到的工作空间大小不足的错误。作者引用了onnx/onnx-tensorrt的issue #758，并指出可以通过增大工作空间大小来解决。在jetson-inference的源代码中，发现针对不同TensorRT版本设置了默认工作空间大小。对于大于TensorRT 5的版本，原本设置为32MB，作者建议将其更改为1GB。修改这一设置可以成功解决转换问题。
关于使用jetson-inference onnx转tensorrt报错，
查到onnx/onnx-tensorrt#758说是要增加工作空间大小 ，但是jetson-inference封装了
阅读代码发现在jetson-inference-master\c\tensorNet.cpp
#if NV_TENSORRT_MAJOR > 5
mWorkspaceSize = 32 << 20;
#else
mWorkspaceSize = 16 << 20;
#endif
将32<<20改为1<<30就可以了
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
