ttanzhiqiang / onnx_tensorrt_project

Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxnet to onnx to tensorrt
210 stars 43 forks source link

yolo5生成失败: #9

Closed xinsuinizhuan closed 3 years ago

xinsuinizhuan commented 3 years ago

reading calib cache: E:\comm_Item\Item_done\onnx_tensorrt_pro\onnx_tensorrt_project-main\model\pytorch_onnx_tensorrt_yolov5\yolov5s.table TensorRT was linked against cuDNN 8.1.0 but loaded cuDNN 8.0.2 Detected 1 inputs and 7 output network tensors. TensorRT was linked against cuDNN 8.1.0 but loaded cuDNN 8.0.2 TensorRT was linked against cuDNN 8.1.0 but loaded cuDNN 8.0.2 Starting Calibration. dog.jpg 0 Calibrated batch 0 in 1.35348 seconds. person.jpg 1 Calibrated batch 1 in 1.3612 seconds. Post Processing Calibration data in 0.0005131 seconds. Calibration completed in 20.8498 seconds. reading calib cache: E:\comm_Item\Item_done\onnx_tensorrt_pro\onnx_tensorrt_project-main\model\pytorch_onnx_tensorrt_yolov5\yolov5s.table Writing Calibration Cache for calibrator: TRT-7203-MinMaxCalibration writing calib cache: E:\comm_Item\Item_done\onnx_tensorrt_pro\onnx_tensorrt_project-main\model\pytorch_onnx_tensorrt_yolov5\yolov5s.table size: 4711 TensorRT was linked against cuDNN 8.1.0 but loaded cuDNN 8.0.2 C:\source\rtSafe\cuda\cudaConvolutionRunner.cpp (483) - Cudnn Error in nvinfer1::rt::cuda::CudnnConvolutionRunner::executeConv: 8 (CUDNN_STATUS_EXECUTION_FAILED) C:\source\rtSafe\cuda\cudaConvolutionRunner.cpp (483) - Cudnn Error in nvinfer1::rt::cuda::CudnnConvolutionRunner::executeConv: 8 (CUDNN_STATUS_EXECUTION_FAILED) [2021-08-08 11:28:54.729] [info] serialize engine to E:\comm_Item\Item_done\onnx_tensorrt_pro\onnx_tensorrt_project-main\model\pytorch_onnx_tensorrt_yolov5\yolov5s_fp32_batch_1.engine [2021-08-08 11:28:54.730] [error] engine is empty, save engine failed [2021-08-08 11:28:54.731] [info] create execute context and malloc device memory... [2021-08-08 11:28:54.731] [info] init engine...

ttanzhiqiang commented 3 years ago

Cudnn Error in nvinfer1::rt::cuda::CudnnConvolutionRunner::executeConv: 8 (CUDNN_STATUS_EXECUTION_FAILED) cudnn error.check you cuda and cudnn