Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Different result in different GPU #107

Open
cashtsangwh opened this issue Dec 7, 2023 · 1 comment
Open

Different result in different GPU #107

cashtsangwh opened this issue Dec 7, 2023 · 1 comment

Comments

@cashtsangwh
Copy link

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

@roboserg
Copy link

Did you figure it out?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants