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

多显卡运行,报错。 #101

Open
xinsuinizhuan opened this issue May 17, 2022 · 2 comments
Open

多显卡运行,报错。 #101

xinsuinizhuan opened this issue May 17, 2022 · 2 comments

Comments

@xinsuinizhuan
Copy link

可能需要多显卡运行,模型1在显卡1上运行,模型2在显卡2上运行。
模型1:
模型转换的时候SimpleYolo::set_device(0),推理的时候commit(_ImgSrc).get();也进行了SimpleYolo::set_device(0),运行正常,

但是模型2要在显卡2上运行时:
模型转换的时候SimpleYolo::set_device:1),推理的时候commit(_ImgSrc).get();也进行了SimpleYolo::set_device(1),崩溃。

@shouxieai
Copy link
Owner

仅仅在create_infer时指定的deviceid有效,其他时候设置没有意义

@shouxieai
Copy link
Owner

并且崩溃的提示是什么呢?

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