x86 下转换的 om 模型在 arm 机器上无法使用 Yolov5_for_Pytorch
收藏回复举报
x86 下转换的 om 模型在 arm 机器上无法使用 Yolov5_for_Pytorch
t('forum.solved') 已解决
新人帖
发表于2024-01-02 14:28:03
0 查看

问题描述

我在使用官方的例子:modelzoo-GPL/ built-in / ACL_Pytorch / Yolov5_for_Pytorch 在一台 atlas 300-3010 机器上(x86 平台)转换 yolov5 模型为 om 模型,该 om 模型可以正常的跑通 Yolov5_for_Pytorch 的示例。我在 atlas500 智慧小站上使用了上面在 x86 上转换的 om 模型继续跑 Yolov5_for_Pytorch 的示例,但是报告如下错误:
infer_errror.png

问题列表:
1.我想要知道上述报错的原因是什么,如何解决。

2.om 模型在 x86-64 和 arm 上面是通用的吗?
希望社区大佬帮助解决,谢谢

环境情况

我的环境为使用docker 启动,启动命令如下:
docker run -it --rm -e LD_LIBRARY_PATH=/home/data/miniD/driver/lib64:$LD_LIBRARY_PATH --device=/dev/davinci0 --device=/dev/davinci_manager --device=/dev/devmm_svm --device=/dev/hisi_hdc -v /mnt/tfcard/lomogo/download/yolov5:/yolov5 -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /home/data/miniD/driver/lib64:/home/data/miniD/driver/lib64 -v /usr/local/Ascend/ascend-toolkit:/usr/local/Ascend/ascend-toolkit yolov5_base:v0.1.0 bash

启动之后使用 npu-smi info 会报错:
npu-smi.png

本帖最后由 匿名用户2024/01/03 10:57:53 编辑

我要发帖子