tensorRT_Pro icon indicating copy to clipboard operation
tensorRT_Pro copied to clipboard

WIN10 CUDA10.2 TRT7.0 出现错误

Open QIANXUNZDL123 opened this issue 2 years ago • 3 comments

我下载了最新的trt_pro的程序,使用了里面的exmple_simple_yolo,编译的时候没有出现问题,可以正常使用,但是使用自己模型就会出现问题: image

我已经导出为动态的bs,pytorch=1.10,请问这是什么原因,我设置了绝对路径也是包这样的错误。。。

QIANXUNZDL123 avatar Nov 15 '22 04:11 QIANXUNZDL123

这是因为resize节点有问题,你需要降低pytorch版本,或者使用simplify过一下

shouxieai avatar Nov 22 '22 13:11 shouxieai

这是因为resize节点有问题,你需要降低pytorch版本,或者使用simplify过一下

好的 ,我试试simplify,但是我在TensorRT主文件夹里面使用app_yolo编译我的模型是没有问题得,很奇怪

QIANXUNZDL123 avatar Nov 23 '22 01:11 QIANXUNZDL123

这是因为resize节点有问题,你需要降低pytorch版本,或者使用simplify过一下

你好,我是使用的tensort7.0版本,我之前的onnx文件已经经过simplify了,报错提示节点有问题,然后我使用提供的脚本进行resize转换后变成了unsample,但onnx文件就无法解析了,请问改怎么处理

lwh1229 avatar Dec 01 '23 08:12 lwh1229