deepinsight / deepinsight/insightface
How to specify gpu
Open
- Dominant language
- Python
- Stars
- 29.7k
- Forks
- 6.1k
- PR merge metrics
- No merged PRs in 30d
Description
```
face_analysis_model_path = "pretrained_models/face_analysis"
face_analysis = FaceAnalysis(
name="",
root=face_analysis_model_path,
providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
)
face_analysis.prepare(ctx_id=2, det_size=(640, 640))
```
gpu0 is used regardless of ctx_id setting
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.