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

[Feature Request] ascend昇腾npu推理例子 #355

Open
zldrobit opened this issue May 9, 2023 · 1 comment
Open

[Feature Request] ascend昇腾npu推理例子 #355

zldrobit opened this issue May 9, 2023 · 1 comment

Comments

@zldrobit
Copy link

zldrobit commented May 9, 2023

请问可以给个ascend昇腾npu推理的例子吗?官方文档只有cpu和cuda的例子。

@pymumu
Copy link
Contributor

pymumu commented May 9, 2023

ascend的和GPU的图是一样的,差别的地方是推理,解码功能单元device需要指定为ascend即可。
模型文件,要转为ascend的。推理功能单元的那个toml配置文件里面,指定ascend。

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