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.

Sign up for GitHub

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

c++版本 #378

Open

lzylzylzy123456 opened this issue Jul 28, 2023 · 1 comment

lzylzylzy123456 commented Jul 28, 2023

貌似C++版本是在cpu上跑，那么能设置在gpu上跑吗？我看开放的接口里面没有gpu，是不是MNN也要编译GPU版本？

Collaborator

tunmx commented Jul 28, 2023 •

edited

Loading

可以在GPU上跑，需要做以下调整：

编译MNN时打开MNN_CUDA
到代码中把mnn推理的backend修改CUDA推理
需要在宿主机提前配置CUDA环境

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment