grounding_dino_demo.py performs wrong when set device to other GPUs.

IDEA-Research / Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything

https://arxiv.org/abs/2401.14159

Apache License 2.0

14.11k stars 1.31k forks source link

grounding_dino_demo.py performs wrong when set device to other GPUs. #475

Open tingxueronghua opened 3 months ago

tingxueronghua commented 3 months ago

As described in the title, if setting the device to "cuda:2" there are no results by dino.

Still not sure where the bug occurs.

rentainhe commented 3 months ago

Yes, this is something wrong with the Deformable Attention operator, you can try CUDA_VISIBLE_DEVICES=2 instead of cuda:2