longzw1997 / Open-GroundingDino

This is the third party implementation of the paper Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection.
MIT License
420 stars 65 forks source link

I need to share multiple GPUs on a server with my colleagues, so how do I specify which GPUs to use for training when I am using them? #66

Open red-gezi opened 6 months ago

jokercao6 commented 6 months ago

insert os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3,4,5,6,7"
in your main