shouxieai / tensorRT_Pro

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

no kernel image is available for execution #162

Closed XDxiaoqi closed 2 years ago

XDxiaoqi commented 2 years ago
direct_test();
或者

test(SimpleYolo::Type::V7, SimpleYolo::Mode::FP32, "yolov7"); 错误如下: [error][simple_yolo.cu:556]:launch failed: no kernel image is available for execution on the device [error][simple_yolo.cu:450]:launch failed: no kernel image is available for execution on the device [error][simple_yolo.cu:454]:launch failed: no kernel image is available for execution on the device 怎么修改

XDxiaoqi commented 2 years ago

如果你是不同显卡,请设置为显卡对应的号码参考这里:https://developer.nvidia.com/zh-cn/cuda-gpus#compute

set(CUDA_GEN_CODE "-gencode=arch=compute_86,code=sm_86") 在cmake 设置一下的