atlas 200dk 上mindspore_server官方案例进行推理失败

在Atlas200DK上安装MindSpore环境并运行成功后,尝试将PC上的MindSpore模型移植过来进行推理。使用mindspore_server时遇到错误,问题源于示例代码中指定了设备ID(0,1),而实际只需要设备0。修改设备IDs为(0,)后,问题得到解决。
摘要由CSDN通过智能技术生成

在atlas 200dk上安装了合设环境,然后安装了mindspore1.6, 验证案例已经成功跑通

然后,我想要把PC机上的mindspore模型移植到atlas 200dk上推理

通过查阅文档,发现可以使用mindspore_server进行(不确定可不可行,因为现在报错了)

然后我开始跑官网的案例

https://www.mindspore.cn/serving/docs/zh-CN/r1.6/serving_example.html

然后报错了

这个是为什么呢

我的tensor_add.mindir是PC机(x86_64)mindspore1.6.2版本导出来的, atlas上为了和mindspore_server保持版本一致用了1.6.0,不知道有没有关系

****************************************************解答*****************************************************

问题就出在这,样例中是使用设备0和1,你需要把设备1删除掉。 serving_server.py中     servable_config = server.ServableStartConfig(servable_directory=servable_dir, servable_name="add",                                                  device_ids=(0, 1)) device_ids改成(0,)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值