我使用这个命令:
trtexec --onnx=solov2.onnx --saveEngine=solov2.trt
报错部分:
[05/06/2022-15:38:59] [E] [TRT] ModelImporter.cpp:776: --- End node ---
[05/06/2022-15:38:59] [E] [TRT] ModelImporter.cpp:779: ERROR: builtin_op_importers.cpp:3352 In function importRange:
[8] Assertion failed: inputs.at(0).isInt32() && "For range operator with dynamic inputs, this version of TensorRT only supports INT32!"
[05/06/2022-15:38:59] [E] Failed to parse onnx file
[05/06/2022-15:38:59] [I] Finish parsing network model
[05/06/2022-15:38:59] [E] Parsing model failed
[05/06/2022-15:38:59] [E] Failed to create engine from model.
[05/06/2022-15:38:59] [E] Engine set up failed
&&&& FAILED TensorRT.trtexec [TensorRT v8204] # trtexec --onnx=solov2.onnx --saveEngine=solov2.trt
-
solov2生成好的onnx模型转tensorrt的时候报错,麻烦看一下谢谢!
-
刘看山 |
如果新版本的tensorrt无法转,可以先用 onnx-simplifier 简化一下看看。
-
@JimXu1989 你好,请问你现在有成功转成trt格式吗?
-
@JimXu1989 请问一下 您使用的是神力版的SOLOv2吗 我用的是AdelaiDet 模型转化不成功
-
刘看山 |
我这边也无法转了。trt更新了接口。
-
@刘看山 在 solov2生成好的onnx模型转tensorrt的时候报错,麻烦看一下谢谢! 中说:
是一下 onnx-tensorrt 这个工具。
你好,我用了你说的这个工具,报错是差不多的:
onnx2trt solov2.onnx -o solov2.trt
Input filename: solov2.onnx
ONNX IR version: 0.0.7
Opset version: 11
Producer name: pytorch
Producer version: 1.10
Domain:
Model version: 0
Doc string:
Parsing model
[2022-05-07 07:05:49 WARNING] onnx2trt_utils.cpp:364: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
While parsing node number 283 [Range -> "966"]:
ERROR: builtin_op_importers.cpp:3350 In function importRange:
[8] Assertion failed: inputs.at(0).isInt32() && "For range operator with dynamic inputs, this version of TensorRT only supports INT32!" -
你好,我用了你说的这个工具,报错是差不多的:
onnx2trt solov2.onnx -o solov2.trtInput filename: solov2.onnx
ONNX IR version: 0.0.7
Opset version: 11
Producer name: pytorch
Producer version: 1.10
Domain:
Model version: 0
Doc string:Parsing model
[2022-05-07 07:05:49 WARNING] onnx2trt_utils.cpp:364: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
While parsing node number 283 [Range -> "966"]:
ERROR: builtin_op_importers.cpp:3350 In function importRange:
[8] Assertion failed: inputs.at(0).isInt32() && "For range operator with dynamic inputs, this version of TensorRT only supports INT32!" -
刘看山 |
是一下 onnx-tensorrt 这个工具。
-
使用tensorrt 8.2 cuda 11.1 pytorch 1.10.1 opencv 4.5.5