You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
First I create a predictor in the following way to get an output:
cfg.MODEL.DEVICE = "cuda:0"
predictor = DefaultPredictor(cfg)
outputs = predictor(image, "panoptic")
Then I want to use the model in second GPU and get the output, I also create it in the following way.
cfg.MODEL.DEVICE = "cuda:1"
predictor = DefaultPredictor(cfg)
outputs = predictor(image, "panoptic")
However, when I compare both of the outputs, the result is totally difference. The output using cuda 0 do the segmentation task very well. But the outputs using cuda 1 cannot recognize any object. I don't know why
The text was updated successfully, but these errors were encountered:
First I create a predictor in the following way to get an output:
cfg.MODEL.DEVICE = "cuda:0"
predictor = DefaultPredictor(cfg)
outputs = predictor(image, "panoptic")
Then I want to use the model in second GPU and get the output, I also create it in the following way.
cfg.MODEL.DEVICE = "cuda:1"
predictor = DefaultPredictor(cfg)
outputs = predictor(image, "panoptic")
However, when I compare both of the outputs, the result is totally difference. The output using cuda 0 do the segmentation task very well. But the outputs using cuda 1 cannot recognize any object. I don't know why
The text was updated successfully, but these errors were encountered: