insightface
insightface copied to clipboard
Unable to use GPU
Already installed onnxruntime-gpu,ctx_id=0,There is also GPU:0 in the environment,But the model still runs using cpu
Please check the compatibility of onnxruntime-gpu library with your GPU model.
Thank you, it is indeed a version mismatch issue
What was the version mismatch issue you had?
I was running with GPU version fine on cloud machine but then one day it starts loading CPU version instead.
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
I'm not sure what exactly is the cause. Tried decreasing onnxruntime-gpu
version from 1.15.0 to 1.14.0 but it doesn't help.
OK In my case it turns out to be an issue relating to importing pytorch. If I don't import torch
before importing FaceAnalysis
then only CPU model would be loaded.
https://github.com/deepinsight/insightface/issues/2344
does anyone please know what to do?
does anyone please know what to do?
uninstall onnxruntime and then install onnxruntime-gpu.Then you can use gpu