ZjczSSKrSSKJr SSKJrJrJrJrJr SSK J r J r SSK J r "SS\5r"S S 5rg) N)Enum)AnyDictListSequenceTuple)get_available_providers get_device)loggerc(\rSrSrSrSrSrSrSrSr g) EP CPUExecutionProviderCUDAExecutionProviderDmlExecutionProviderCANNExecutionProviderCoreMLExecutionProviderN) __name__ __module__ __qualname____firstlineno__CPU_EPCUDA_EP DIRECTML_EPCANN_EP COREML_EP__static_attributes__rr/home/mande/repo/quber/.venv/lib/python3.13/site-packages/rapidocr/inference_engine/onnxruntime/provider_config.pyrr s #F%G(K%G)Ir rc,\rSrSrS\\\44SjrS\\ \\\\444Sjr S\\\44Sjr S\\\44Sjr S\\\44Sjr S\\\44S jrS\\\44S jrS \\4S jrS\4S jrS\4SjrS\4SjrS\4SjrS\\4SjrSrg)ProviderConfig engine_cfgc[5UlURSUlURSS5UlURSS5UlURSS5UlURSS5UlXlg)Nruse_cudaFuse_dmluse_cann use_coreml) r had_providersdefault_providerget cfg_use_cuda cfg_use_dml cfg_use_canncfg_use_coremlcfg)selfr%s r!__init__ProviderConfig.__init__sq(?(A $ 2 21 5&NN:u=%>>)U;&NN:u=(nn\5Ar returncX[RRUR54/nUR 5(a:UR S[R RUR545 UR5(aP[R"S5 UR S[RRUR545 UR5(aP[R"S5 UR S[RRUR545 UR!5(aP[R"S5 UR S[R"RUR%545 U$)NrzEWindows 10 or above detected, try to use DirectML as primary providerz)Try to use CANNExecutionProvider to inferz9macOS/iOS detected, try to use CoreML as primary provider)rrvalue cpu_ep_cfgis_cuda_availableinsertr cuda_ep_cfgis_dml_availabler infor dml_ep_cfgis_cann_availabler cann_ep_cfgis_coreml_availabler coreml_ep_cfg)r3resultss r! get_ep_listProviderConfig.get_ep_list!s IIOOT__%678  ! ! # # NN1rzz//1A1A1CD E  " " KKW  NN1r~~33T__5FG H  ! ! # # KKC D NN1rzz//1A1A1CD E  # # % % KKS T NN1r||1143E3E3GH Ir c@[URR5$N)dictr2r9r3s r!r9ProviderConfig.cpu_ep_cfg7sDHH''((r c@[URR5$rH)rIr2r<rJs r!r<ProviderConfig.cuda_ep_cfg:DHH(())r cURRbURR$UR5(aUR5$UR 5$rH)r2r?r:r<r9rJs r!r?ProviderConfig.dml_ep_cfg=sM 88   *88&& &  ! ! # ###% %  r c@[URR5$rH)rIr2rArJs r!rAProviderConfig.cann_ep_cfgErNr c@[URR5$rH)rIr2rCrJs r!rCProviderConfig.coreml_ep_cfgHsDHH**++r session_providerscU(d [S5eUSn[RUR[RUR [R UR[RUR0nUR5HcupEU"5(dMX$R:wdM$[R"URSUS35 [R"SU35 Me g)NzSession Providers is emptyrzT is available, but the inference part is automatically shifted to be executed under z. zThe available lists are ) ValueErrorrrr:rr=rr@rrBitemsr8r warning)r3rUfirst_providerproviders_to_checkep check_funcs r!verify_providersProviderConfig.verify_providersKs 9: :*1- JJ.. NND11 JJ.. LL$22  1668NB||(( :xxj tvDuEEGH!9:K9LMN 9r c@UR(dg[RRn[ 5S:XaXR ;ag[ R"USUR SURS35 SUS3S S S S S US3/nURU5 g)NFGPUT is not in available providers (). Use  inference by default.zIf you want to use za acceleration, you must do:(For reference only) If you want to use GPU acceleration, you must do:AFirst, uninstall all onnxruntime packages in current environment.zASecond, install onnxruntime-gpu by `pip install onnxruntime-gpu`.zHNote the onnxruntime-gpu version must match your cuda and cudnn version.zdYou can refer this link: https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.htmlThird, ensure zW is in available providers list. e.g. ['CUDAExecutionProvider', 'CPUExecutionProvider']) r.rrr8r r+r rYr, print_log)r3rinstall_instructionss r!r: ProviderConfig.is_cuda_available_s  **"" <5 W0B0B%Bi78J8J7K7SWShShRii A "'+U U O O V rWI%| }  +,r cUR(dg[R"5nUS:wa([R"SUSUR S35 g[R "5RS5SnUR5(a [U5OSnUS :a([R"S USUR S35 g[RRnX@R;ag [R"US URS UR S35 SSSSUS3/nURU5 g)NFWindowszFirst, ensure you have installed Huawei Ascend software stack.zOSecond, install onnxruntime with CANN support by following the instructions at:za https://onnxruntime.ai/docs/execution-providers/community-maintained/CANN-ExecutionProvider.htmlrfzW is in available providers list. e.g. ['CANNExecutionProvider', 'CPUExecutionProvider']) r0rrr8r+r rYr,rg)r3rrhs r!r@ ProviderConfig.is_cann_availables  **"" (( (i78J8J7K7SWShShRii A  A L ] pWI%| }   +,r cUR(dg[R"5nUS:wa([R"SUSUR S35 g[ RRnX R;ag[R"USURSUR S35 S S US 3/nURU5 g) NFDarwinz9CoreML is only supported in macOS/iOS. The current OS is rlrdTrbrczCThe standard onnxruntime package for macOS includes CoreML support.zEnsure zY is in available providers list. e.g. ['CoreMLExecutionProvider', 'CPUExecutionProvider']) r1rorpr rYr,rrr8r+rg)r3rurrhs r!rB"ProviderConfig.is_coreml_availables""" X  NNKF8SYZ^ZoZoYpqGH LL&& ** *k9$:L:L9MWUYUjUjTklB C  Ri[ y z  +,r log_listcBUHn[R"U5 M grH)r r>)r3rlog_infos r!rgProviderConfig.print_logs H KK !!r )r2r0r1r.r/r,r+N)rrrrrstrrr4rrrEr9r<r?rArCrr^boolr:r=r@rBrgrrr r!r#r#s 4S> T%T#s(^(;"<=,)DcN)*T#s(^*!DcN!*T#s(^*,tCH~,O(3-O(4.#$#J4*T2"$s)"r r#)roenumrtypingrrrrr onnxruntimer r utils.logr rr#rr r!rs133;**v"v"r