请问如何配置才能使用gpu？

[big-mango-88]

发现没用上GPU，已经根据 rapidocr 进行了如下配置：
pdf_extractor = PDFExtracter(
ocr_params={
"Global.with_onnx": True,
"EngineConfig.onnxruntime.use_cuda": True
}
)

[SWHL]

用torch版的GPU，不推荐用onnx的GPU

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 发送日期 | 2025年05月02日 23:37 | | 收件人 | RapidAI/RapidOCRPDF ***@***.***> | | 抄送人 | Subscribed ***@***.***> | | 主题 | [RapidAI/RapidOCRPDF] 请问如何配置才能使用gpu？ (Discussion #26) | 发现没用上GPU，已经根据 rapidocr 进行了如下配置： pdf_extractor = PDFExtracter( ocr_params={ "Global.with_onnx": True, "EngineConfig.onnxruntime.use_cuda": True } ) — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: ***@***.***>

[SWHL]

确认一下torch GPU是否可以正常使用

[SWHL]

注意PDF要是扫描版的。如果是可以直接提取内容的PDF，这个就不走GPU了。

[SWHL]

参考一下官方文档

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 发送日期 | 2025年05月04日 09:42 | | 收件人 | RapidAI/RapidOCRPDF ***@***.***> | | 抄送人 | SWHL ***@***.***>, Comment ***@***.***> | | 主题 | Re: [RapidAI/RapidOCRPDF] 请问如何配置才能使用gpu？ (Discussion #26) | 使用如下 torch 设置也没用到 GPU，请问是配置问题还是依赖安装的不对？ pdf_extractor = PDFExtracter( ocr_params={ # "Global.with_onnx": True, # "EngineConfig.onnxruntime.use_cuda": True "Global.with_torch": True, "EngineConfig.torch.use_cuda": True } ) — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: ***@***.***>