search

CaoWGG / TensorRT-YOLOv4

tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn
264 stars 76 forks source link

ERROR: Unable to create engine #56

Open wkal opened 3 years ago

wkal commented 3 years ago

@CaoWGG How to fix this error? Platform information: Python: Python3.8 Cuda: V10.0.130 GPU: RTX3090 gcc: 7.5.0

Runing command: ./buildEngine -i model/yolov4.onnx -o model/yolov4.engine -b 1 -m 0

Runing log: input-onnx: model/yolov4.onnx output-engine: model/yolov4.engine max-batchsize: 1 calib-file:

Input filename: model/yolov4.onnx ONNX IR version: 0.0.4 Opset version: 9 Producer name: https://github.com/CaoWGG Producer version: Domain:
Model version: 0 Doc string:

WARNING: ONNX model has a newer ir_version (0.0.4) than this parser was built against (0.0.3). Begin building engine... ERROR: CUDA cask failure at execution for trt_volta_scudnn_winograd_128x128_ldg1_ldg4_relu_tile148t_nt_v1. ERROR: cuda/caskConvolutionLayer.cpp (256) - Cuda Error in execute: 48 ERROR: cuda/caskConvolutionLayer.cpp (256) - Cuda Error in execute: 48 ERROR: Unable to create engine