tensorRT_Pro
tensorRT_Pro copied to clipboard
pytorch版本的export()的onnx导出问题
老师,您好!我在windows下配置了pytorch1.11.0,在对自定义插件进行export()时,会报异常
查看原生函数接口,发现在1.11.0版本下enable_onnx_checker并不在export()的参数列表中。请问在不换pytorch版本的情况下有没有什么好的解决方法。
同学你好!我查看了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: @.***>