IT博客汇
  • 首页
  • 精华
  • 技术
  • 设计
  • 资讯
  • 扯淡
  • 权利声明
  • 登录 注册

    在CANN推理程序中,执行aclmdlExecute()函数失败。 返回错误码:507011

    小王发表于 2024-04-16 12:42:00
    love 0
    1、在yolov7中把yolov7.pt转为.onnx
    python3 export.py --weights weights/yolov7.pt --grid --simplify --topk-all 100 --img-size 640 640 --max-wh 640

    2、在CANN环境下使用act把.onnx转为.om
    atc --model=yolov7.onnx --framework=5 --output=yolov7 --input_shape="images:1,3,640,640"  --soc_version=Ascend310  --insert_op_conf=aipp.cfg

    3、在C++程序中使用CANN对yolov7.om模型推理,执行aclmdlExecute()函数失败。
    返回错误码:507011

    解决办法:
    在阿特拉斯设备中查看设备信息:
    npu-smi info
    可以看到芯片型号是310B1

    修改act命令参数:
    atc --model=yolov7.onnx --framework=5 --output=yolov7 --input_shape="images:1,3,640,640"  --soc_version=Ascend310B1  --insert_op_conf=aipp.cfg

    把生成的.om文件放到C++程序中进行推理,推理成功!!!



    O了!
    我翘里哇!!!!!!!!!!!!!!!!!!!!!!!!!!


    小王 2024-04-16 20:42 发表评论


沪ICP备19023445号-2号
友情链接