8.1.2 算子集版本 (Opset Version) 兼容性管理


文档摘要

8.1.2 算子集版本 (Opset Version) 兼容性管理 在深度学习模型部署的现实战场上,我们常常遭遇这样一幕:一个在 PyTorch 1.13 + ONNX opset 17 下训练并导出的 ResNet-50 模型,被小心翼翼地送入 TensorRT 8.6 推理引擎——结果却在 阶段报出刺眼的错误: 。而此时,你刚刚确认过模型里确实只用了 ,且 PyTorch 文档清清楚楚写着“自 1.12 起支持导出为 Round 算子”。你翻遍 TensorRT 的 Release Notes,发现它直到 8.6 GA 才正式支持 opset 18 中新增的 (v18),但默认仍以 opset 17 兼容模式加载 ONNX。你不是漏了什么配置?


发布者: 作者: 转发
评论区 (0)
U