When I set CUDA_VISIBLE_DEVICES="1" not "0"，The Code dose not work well.

[KyanChen]

I found when I set CUDA_VISIBLE_DEVICES="1", The Code will terminal at self.model = CustomCLIP(cfg, classnames, clip_model) self.model.to(self.device).

The to(self.device) function will wait a long time, and will not step over.

[Kyrie10favor]

I solved it this way. Original code

self.model.to(self.device)

New Code

device_ids = [4,5,6,7]
self.device = device_ids[0]
self.model.to(self.device)