shouxieai / tensorRT_Pro

C++ library based on tensorrt integration
MIT License
2.62k stars 549 forks source link

在提供的docker中运行案例时出现问题 #122

Closed Yuhz288 closed 2 years ago

Yuhz288 commented 2 years ago

显卡是3060,安装了nvidia-container-runtime,并修改了docker的项目的cuda_arch。在运行make yolo -j64有如下报错,请问该如何修改?

[2022-07-20 12:26:46][error][yolo_decode.cu:115]:launch failed: no kernel image is available for execution on the device [2022-07-20 12:26:46][error][yolo_decode.cu:122]:launch failed: no kernel image is available for execution on the device [2022-07-20 12:26:46][error][yolo_decode.cu:115]:launch failed: no kernel image is available for execution on the device [2022-07-20 12:26:46][error][yolo_decode.cu:122]:launch failed: no kernel image is available for execution on the device [2022-07-20 12:26:46][error][yolo_decode.cu:115]:launch failed: no kernel image is available for execution on the device [2022-07-20 12:26:46][error][yolo_decode.cu:122]:launch failed: no kernel image is available for execution on the device [2022-07-20 12:26:46][info][app_yolo.cpp:83]:yolov5m.FP32.trtmodel[YoloV5] average: 9.34 ms / image, FPS: 107.07 [2022-07-20 12:26:46][info][app_yolo.cpp:109]:Save to yolov5m_YoloV5_FP32_result/car.jpg, 0 object, average time 9.34 ms [2022-07-20 12:26:46][info][app_yolo.cpp:109]:Save to yolov5m_YoloV5_FP32_result/zgjr.jpg, 0 object, average time 9.34 ms [2022-07-20 12:26:46][info][app_yolo.cpp:109]:Save to yolov5m_YoloV5_FP32_result/yq.jpg, 0 object, average time 9.34 ms [2022-07-20 12:26:46][info][app_yolo.cpp:109]:Save to yolov5m_YoloV5_FP32_result/zand.jpg, 0 object, average time 9.34 ms [2022-07-20 12:26:46][info][app_yolo.cpp:109]:Save to yolov5m_YoloV5_FP32_result/gril.jpg, 0 object, average time 9.34 ms [2022-07-20 12:26:46][info][app_yolo.cpp:109]:Save to yolov5m_YoloV5_FP32_result/group.jpg, 0 object, average time 9.34 ms [2022-07-20 12:26:46][info][yolo.cpp:287]:Engine destroy.

Yuhz288 commented 2 years ago

原环境是3060+ubuntu20.04+cuda11.6

shouxieai commented 2 years ago

先执行make clean,然后再make run,不小心残留的文件,不好意思