C++ uses GPU | 在C++中使用GPU
KunMengcode opened this issue · 0 comments
KunMengcode commented
In YOLOV7_face Class Add the following code to the initialization location
在 YOLOV7_face 类中,初始化位置添加如下代码
OrtSessionOptionsAppendExecutionProvider_CUDA(sessionOptions, 0);