专栏算法工具链尝试使用GPU加速ONNX推理失败

尝试使用GPU加速ONNX推理失败

已解决
DZone2024-03-13
105
3

用户您好,请详细描述您所遇到的问题,详细的描述有助于帮助我们快速定位,解决问题~Thanks♪(・ω・)ノ

1. 芯片型号:J5

2. 天工开物开发包 OpenExplorer 版本:例如 J5_OE_1.1.68

3. 问题定位:模型HBONNX测试

4. 问题具体描述:

尝试使用论坛回复的GPU加速方法失败,程序卡死。

运行的代码路径:/open_explorer/ddk/samples/ai_toolchain/horizon_model_convert_sample/03_classification/01_mobilenet/mapper# sh 04_inference.sh

运行环境:docker版本和oe包版本:docker_open_explorer_ubuntu_20_j5_gpu_v1.1.68.tar.gz horizon_j5_open_explorer_v1.1.68-py38_20231014.tar.gz

运行状态如下图:cpu一直在占用,程序一直是卡死状态


5. 如果提供的信息过多,建议进行打包并上传附件至百度网盘

算法工具链
征程5
评论3
0/1000
  • J6标定问题
    Lv.1

    1.您稍等,我复现一下您的这个问题

    2024-03-13
    0
    0
  • DZone
    Lv.1

    查看帖子发现图像加载不出来,这边补充以上提到的两张图片。

    2024-03-13
    0
    0
  • J6标定问题
    Lv.1

    您好,我复现确实存在这样的问题,现在只支持cuda-10.0,并且显卡不支持RTX 30系列和RTX40系列,并且我们这边在RTX2080验证过,可以行。显卡要符合要求,重点是cuda10,显卡要支持cuda10。

    2024-03-15
    0
    0