使用`$ 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
breezedeus commentedon Feb 12, 2025
😅 这坑好深啊。。
breezedeus commentedon Feb 12, 2025
rapidocr_onnxruntime 是不是默认装了 onnxruntime 的CPU版本。可以试试能不能先装 rapidocr_onnxruntime,然后卸载掉 onnxruntime,然后重装 onnxruntime-gpu ?
lacser commentedon Feb 14, 2025
确实是这样的,当时我也发现了,但是老版本的rapidocr_onnxruntime非要默认装老版本的onnxruntime,卸载了就报错