deepinsight / deepinsight/insightface

Multi-gpu usage

Open
#1,116 7 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29.7k
Forks
6.1k
PR merge metrics
No merged PRs in 30d

Description

model = insightface.app.FaceAnalysis()
ctx_id = 0
model.prepare(ctx_id = ctx_id, nms=0.4)

This is how i build model. with ctx_id param i can select only one gpu for model,

how can i choose all 4 gpus for model, is there any way ? putting in list [0,1,2,3] does not work btw.

Thanks!

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.