o H!i@sddlZddlZddlZddlZddlZddlZddlmZddl m Z e Z ddZ ddZd d Zd d Zdd dZdddZdddZddZejddZejddZdS)N)logging)get_device_typecOstj|i|SN) subprocess check_call)argskwargsr f/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/repo_manager/utils.py _check_callsr csddlfdd}||}||}t||D]+\}}|dur$|dur$q|dus0|dur3||kr3dS|dus?|durB||krBdSqdS)NrcsVd}||}|std|t|dt|dt|d|dfS)Nzf^(?P\d+)\.(?P\d+)\.(?P\d+)(?:-(?P.*))?(?:\+(?P.+))?$zUnexpected version string: majorminorpatchZ pre_release)compilematch ValueErrorintgroup) version_strZversion_patternrrer r parse_version#s    z'_compare_version..parse_version)rzip)Zversion1Zversion2rZv1_infosZv2_infosZv1_infoZv2_infor rr _compare_version s rcCs,z tj|WdStjjyYdSw)NFT) importlibmetadata distributionZPackageNotFoundError)packager r r check_package_installation=s r!cCsddl}dd}|dkr`tjtj|ddrb tdkrYt|d dkrYd tvrY|rY| sYt tj|ddt j d d g}t |WddS1sRwYdStd dSdSdS)z-install paddle repository custom dependenciesrNcSstddgddS)NZgccz --versionzutf-8r)r check_outputdecodesplitr r r r get_gcc_versionIsz.install_external_deps..get_gcc_versionZPaddleDetectionZppdetZext_opLinuxz8.2.0Zgpuzsetup.pyinstallaUThe custom operators in PaddleDetection for Rotated Object Detection is only supported when using CUDA, GCC>=8.2.0 and Paddle>=2.0.1, your environment does not meet these requirements, so we will skip the installation of custom operators under PaddleDetection/ppdet/ext_ops, which means you can not train the Rotated Object Detection models.)paddleospathexistsjoinPLATFORMrrZis_compiled_with_cudaZis_compiled_with_rocmswitch_working_dirsys executabler rwarning) repo_name repo_rootr(r%rr r r install_external_depsEs*   "r4cCs@gd}t|tr |g}|||dur|d|gt|S)clone_repo_using_git)gitclone--depth1Nz-b) isinstancestrextendr )urlbranchrr r r r5ds  r5rcCs dd||dt|g}t|dS)fetch_repo_using_gitr6fetchr8N)r;r )r>r=depthrr r r r?os r?TcCsddd|g}t|S)reset_repo_using_gitr6resetz--hard)r )ZpointerZhardrr r r rBus rBcCs:tj|rtdkrtddd|gddStdd|gSd S) remove_repo_using_rmWindowsrmdirz/Sz/QT)shellZrmz-rfN)r)r*r+r-r )namer r r rD{s rDc csttjd>}t| t| dVWdn1s!wYWdn1s0wYWddSWddS1sHwYdS)mutewN)openr)devnull contextlibredirect_stdoutredirect_stderr)fr r r rIs"rIc cs8t}t|z dVWt|dSt|w)r.N)r)getcwdchdir)Znew_wdcwdr r r r.s  r.r)r)T)rMimportlib.metadatarr)platformrr/utilsrZ utils.envrsystemr-r rr!r4r5r?rBrDcontextmanagerrIr.r r r r s*