-
是否可以添加TensorrtExecutionProvider和ROCMExecutionProvider? |
Beta Was this translation helpful? Give feedback.
Answered by
SWHL
Oct 10, 2023
Replies: 2 comments
-
后续会考虑把这块改为参数形式开放到config.yaml中,便于大家自动配置 |
Beta Was this translation helpful? Give feedback.
0 replies
-
我思考再三,发现除 如果需要,可先pip安装 |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
SWHL
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
我思考再三,发现除
CPUExecutionProvider
和CUDAExecutionProvider
比较通用以外,其他的Provider比较小众,暂时先不把上述提到的Provider加到库里了。如果需要,可先pip安装
rapidocr_onnxruntime
,再修改以下位置源码即可。RapidOCR/python/rapidocr_onnxruntime/utils.py
Line 56 in 96c0ec5