search

yuwenmichael / Grounding-DINO-Batch-Inference

Support batch inference of Grounding DINO. "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
2 stars 0 forks source link

issues using cuda #3

Closed solidstudio closed 1 month ago

solidstudio commented 8 months ago

Hi does this work with cuda? I'm using:

https://github.com/HumanSignal/label-studio-ml-backend/tree/master/label_studio_ml/examples/grounding_dino

and getting the following:

label-studio-ml-backend | /app/Grounding-DINO-Batch-Inference/GroundingDINO/groundingdino/models/GroundingDINO/ms_deform_attn.py:31: UserWarning: Failed to load custom C++ ops. Running on CPU mode Only! label-studio-ml-backend | warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only!")

and

output = _C.ms_deform_attn_forward(\\nNameError: name \'_C\' is not defined

yuwenmichael commented 8 months ago

Yes, it should work on cuda. Please refers here by modifying cpu to cuda.