请问如何设置使用GPU而不是CPU推理？

UNeedCryDear / yolov5-opencv-dnn-cpp

使用opencv模块部署yolov5-6.0版本

Apache License 2.0

133 stars 38 forks source link

Closed CaptainEven closed 2 years ago

UNeedCryDear commented 2 years ago

bool Yolo::readModel(Net& net, string& netPath, bool isCuda = false) 读取模型的时候将cuda参数设置为true，前提是你的opencv需要编译过cuda，不然会自动默认切换到cpu运行

CaptainEven commented 2 years ago

@UNeedCryDear 好的，了解。再问一下，如果要使用带CUDA的openCV是需要自己CMake重新编译源代码吗？

UNeedCryDear commented 2 years ago

是的

CaptainEven commented 2 years ago

@UNeedCryDear 好的，多谢！