search

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.19k stars 316 forks source link

error building TensorRT Engine #126

Open rpaaron opened 3 years ago

rpaaron commented 3 years ago

When running ./yolo-trt, I receive:

Building the TensorRT Engine...
ERROR: ../rtSafe/cuda/cutensorReformat.cpp (352) - cuTensor Error in executeCutensor: 18 (Internal cuTensor reformat failed)
ERROR: ../rtSafe/cuda/cutensorReformat.cpp (352) - cuTensor Error in executeCutensor: 18 (Internal cuTensor reformat failed)

Weights file from https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights CUDA 11.0.3 TensorRT 7.2.1.6 Within docker nvidia runtime. Quadro K4200