insightface icon indicating copy to clipboard operation
insightface copied to clipboard

scrfd.py onnx inference doesn't use GPU

Open Nuwan1654 opened this issue 2 years ago • 3 comments

Hi, when I used to run the ONNX converted model using scrfd.py, it seems like using the CPU not the GPU

Nuwan1654 avatar Aug 31 '22 14:08 Nuwan1654

You have to install onnxruntime-gpu library.

nttstar avatar Aug 31 '22 14:08 nttstar

You have to install onnxruntime-gpu library.

it is alredy installed, my onnxruntime version is 1.7.0 and onnxruntime-gpu version is 1.9.0.

Nuwan1654 avatar Aug 31 '22 14:08 Nuwan1654

remove both of them, then re-install onnxruntime-gpu

nttstar avatar Aug 31 '22 15:08 nttstar