Skip to content

要不要支持下gpu解析 #129

Description

@priest2018

感觉cpu一下就炸了

Activity

  1. Joker1212 commented on Mar 11, 2025

    @Joker1212
    Member

    正在写,但是onnx的gpu支持很差其实,启动的时候资源占用会比较多

  2. zzk2021 commented on Mar 27, 2025

    @zzk2021

    正在写,但是onnx的gpu支持很差其实,启动的时候资源占用会比较多

    可以转tensorrt吗?我的场景涉及到大量并发,CPU容易成为瓶颈,所以考虑用GPU,我正在优化代码。

  3. Joker1212 commented on Mar 31, 2025

    @Joker1212
    Member

    正在写,但是onnx的gpu支持很差其实,启动的时候资源占用会比较多

    可以转tensorrt吗?我的场景涉及到大量并发,CPU容易成为瓶颈,所以考虑用GPU,我正在优化代码。

    欢迎提供pr贡献哈,理论上onnx转tensorrt也需要看算子的兼容性

  4. yangailing commented on Mar 31, 2025

    @yangailing

    大佬现在当前表格识别是不是不支持GPU呀?

  5. zzk2021 commented on Apr 15, 2025

    @zzk2021

    正在写,但是onnx的gpu支持很差其实,启动的时候资源占用会比较多

    可以转tensorrt吗?我的场景涉及到大量并发,CPU容易成为瓶颈,所以考虑用GPU,我正在优化代码。

    欢迎提供pr贡献哈,理论上onnx转tensorrt也需要看算子的兼容性

    表格识别的模型有没有pth呀?onnxruntime在创建大量分布在所有可用 GPU 设备上的 ONNX Runtime 会话时候,会有BUG,设置了device id实际全在一个device id上,导致OOM。

  6. Joker1212 commented on Apr 16, 2025

    @Joker1212
    Member

    表格模型都是onnx哈,另外一个库rapidTable有pytorch的模型

  7. CatchBeliF-TAT commented on Apr 16, 2025

    @CatchBeliF-TAT

    麻烦问下现在可以支持GPU推理了吗?感谢

  8. Joker1212 commented on Apr 16, 2025

    @Joker1212
    Member

    麻烦问下现在可以支持GPU推理了吗?感谢

    gpu的话,应该可以用,但是onnx的gpu推理其实也没有明显提速的,可以试试哈,需要安装onnxruntime-gpu吧,参数上也要指定使用gpu

  9. CatchBeliF-TAT commented on Apr 17, 2025

    @CatchBeliF-TAT

    麻烦问下现在可以支持GPU推理了吗?感谢

    gpu的话,应该可以用,但是onnx的gpu推理其实也没有明显提速的,可以试试哈,需要安装onnxruntime-gpu吧,参数上也要指定使用gpu
    用了GPU推理单张4s多,请问是哪里有问题吗?麻烦您啦
    Image

  10. zzk2021 commented on Apr 17, 2025

    @zzk2021

    麻烦问下现在可以支持GPU推理了吗?感谢

    gpu的话,应该可以用,但是onnx的gpu推理其实也没有明显提速的,可以试试哈,需要安装onnxruntime-gpu吧,参数上也要指定使用gpu
    用了GPU推理单张4s多,请问是哪里有问题吗?麻烦您啦
    Image

    我用起来确实加速了,只不过收益不大,后续请关注我的RapidOCR-pytorch项目,使用pth模型和pytorch支持,接口和RapidOCR几乎一样,且误差和paddle版本,onnx版本差别很小

  11. locked and limited conversation to collaborators on Apr 26, 2025
  12. converted this issue into a discussion #142 on Apr 26, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions