tensorRT_Pro icon indicating copy to clipboard operation
tensorRT_Pro copied to clipboard

pytorch版本的export()的onnx导出问题

Open junwei1999 opened this issue 3 years ago • 1 comments

老师,您好!我在windows下配置了pytorch1.11.0,在对自定义插件进行export()时,会报异常 image 查看原生函数接口,发现在1.11.0版本下enable_onnx_checker并不在export()的参数列表中。请问在不换pytorch版本的情况下有没有什么好的解决方法。

junwei1999 avatar Sep 02 '22 03:09 junwei1999

同学你好!我查看了1.11.0中的export函数,参数列表与1.8.0的不同,所以是不支持此参数的传递的。 我目前也没有找到合适的解决办法,你可以更换tensoRT版本为1.8.0试试。 如果你有其他解决方案还请与我联系,共同进步。

------------------ 原始邮件 ------------------ 发件人: "shouxieai/tensorRT_Pro" @.>; 发送时间: 2022年9月12日(星期一) 晚上9:15 @.>; @.@.>; 主题: Re: [shouxieai/tensorRT_Pro] pytorch版本的export()的onnx导出问题 (Issue #147)

同学您好!我也遇到您这个问题了,请问您找到合适的解决办法了嘛~请赐教!

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

junwei1999 avatar Oct 11 '22 07:10 junwei1999