Atlas500 docker镜像内无法调用npu进行推理
收藏回复举报
Atlas500 docker镜像内无法调用npu进行推理
t('forum.solved') 已解决
新人帖
发表于2024-06-22 22:05:01
0 查看
从https://ascendhub.huawei.com/#/detail/infer-modelzoo拉取了镜像运行,在镜像中输入npu-smi info可以检测到310芯片,但是参考https://github.com/mindspore-lab/mindyolo/tree/master部署YOLO时,无法使用npu进行推理,只能使用CPU,原因可能是mindspore只安装在了CPU上,但是换了很多版本都无法安装在Ascend上。请问有解决方法吗,或者有没有针对Atlas500应用部署流程的文档可以供学习参考?

我要发帖子