Grounded-Segment-Anything icon indicating copy to clipboard operation
Grounded-Segment-Anything copied to clipboard

Published 20 hours ago • verified publisher icon IDEA-Research

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

Open tingxueronghua opened this issue 11 months ago • 1 comments

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.

tingxueronghua avatar Mar 18 '24 07:03 tingxueronghua

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

rentainhe avatar Mar 18 '24 15:03 rentainhe