Skip to content

GPU推理问题 #353

Open
Open

Description

@lacser

使用$ pip install cnocr[ort-gpu] 命令安装并没有办法正常调用GPU进行识别。
使用上面的命令安装后,仍然会调用CPU进行识别。经测试,一旦安装1.3.25版本(<1.4的最新版)的rapidocr_onnxruntime后,onnxruntime就无法识别到GPU设备了。并且无论是新老版本的CUDA(11.x~12.x),都无法被正常识别调用。但如果卸载rapidocr_onnxruntime,并重新安装最新版的onnxruntime-gpu,则可以正常识别到GPU。

Activity

  1. breezedeus commented on Feb 12, 2025

    @breezedeus
    Owner

    😅 这坑好深啊。。

  2. breezedeus commented on Feb 12, 2025

    @breezedeus
    Owner

    rapidocr_onnxruntime 是不是默认装了 onnxruntime 的CPU版本。可以试试能不能先装 rapidocr_onnxruntime,然后卸载掉 onnxruntime,然后重装 onnxruntime-gpu ?

  3. lacser commented on Feb 14, 2025

    @lacser
    Author

    确实是这样的,当时我也发现了,但是老版本的rapidocr_onnxruntime非要默认装老版本的onnxruntime,卸载了就报错

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

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions