search

deepinsight / insightface

State-of-the-art 2D and 3D Face Analysis Project
https://insightface.ai
22.83k stars 5.34k forks source link

How to specify gpu #2628

Open wzr0108 opened 1 month ago

wzr0108 commented 1 month ago 
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

ZubairKhan001 commented 1 month ago

face_analysis = FaceAnalysis( name="", root=face_analysis_model_path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"], provider_options=[{"device_id": device_id])

wzr0108 commented 1 month ago 
device_id=5

face_analysis_model_path = "pretrained_models/face_analysis"
face_analysis = FaceAnalysis(
    name="",
    root=face_analysis_model_path,
    providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
    provider_options=[{"device_id": device_id}]
)
face_analysis.prepare(ctx_id=device_id, det_size=(640, 640))

still using gpu0

ZubairKhan001 commented 1 month ago

when the model is loaded what is the value assigned to device id, 0 or 5?

wzr0108 commented 4 weeks ago

i set device_id=5, but the model is loaded into gpu0