Fix cudaError_t error

2023-02-02 09:56:33 -03:00
parent 70c18a72a6
commit cd3ab90867
1 changed files with 1 additions and 0 deletions
--- a/nvdsinfer_custom_impl_Yolo/calibrator.h
+++ b/nvdsinfer_custom_impl_Yolo/calibrator.h
@@ -7,6 +7,7 @@
 #define CALIBRATOR_H

 #include <vector>
+#include <cuda_runtime_api.h>

 #include "NvInfer.h"
 #include "opencv2/opencv.hpp"