o 0 i, @svdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z gZ gZ gZ iZddZddZddZe ed e ed e ed <d d ZddZddZe ed e ed e ed<ddZddZe edddddde ed ddddde ed ddddd e ed!<d"d#Zd$d%Zd&d'Zd(d)Zed*kreejd+ddSdS),zM CUDA Library Installer Installs the latest CUDA library supported by CuPy. NcCsd|d|S)NzAhttps://developer.download.nvidia.com/compute/cudnn/redist/cudnn//platformfilenamerrg/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupyx/tools/install_library.py_make_cudnn_urls rcs\ddgd}|td|fdd|Ddtd|fd d|Ddd d S) N.r)Z _ops_inferZ _ops_trainZ _cnn_inferZ _cnn_trainZ _adv_inferZ _adv_trainz linux-x86_64csg|] }d|dqS)Zlibcudnnz.so.r.0suffix)public_versionrr 2sz'__make_cudnn_record..url filenameszwindows-x86_64csg|] }d|ddqS)cudnnZ64_z.dllrr ) major_versionrrr7s Linux:x86_64zWindows:x86_64)cudarassets)splitr) cuda_versionrfilename_linuxfilename_windowsZ suffix_listr)rrr__make_cudnn_record$s   rcCs<t|dd}|dvsJt|dd|dd|dS) Nr r) z8.8.1zcudnn-linux-x86_64-8.8.1.3_cudaz-archive.tar.xzz!cudnn-windows-x86_64-8.8.1.3_cudaz -archive.zip)intrr)rZ cuda_majorrrr_make_cudnn_record>s   r!z12.xz11.xrcCdd|d|S)Nz7https://developer.download.nvidia.com/compute/cutensor/zredist/libcutensor/z-x86_64/rrrrr_make_cutensor_urlMr#cCs6|||td|d|gdtd|dgdddS)Nlinuxzlibcutensor.so.{}rwindowsz cutensor.dllr)rcutensormin_pypi_versionr)r#format)rrr(rrrrr__make_cutensor_recordTs r*cCst|ddddS)Nz2.1.0z2.0.0z/libcutensor-linux-x86_64-2.1.0.9-archive.tar.xzz.libcutensor-windows-x86_64-2.1.0.9-archive.zip)r*)rrrr_make_cutensor_recordhs r+r'cCr")Nz:https://developer.download.nvidia.com/compute/redist/nccl/vrr)rrrrr_make_nccl_urlxr$r-cCs<|||t||d|gdt||d|gdddS)Nz libnccl.so.{}r)rz Linux:aarch64)rncclr(r)r-r))rZ full_versionrr(Zfilename_linux_x86_64Zfilename_linux_aarch64rrr_make_nccl_records  r/z13.xz2.27.7z!nccl_2.27.7-1+cuda13.0_x86_64.txzz"nccl_2.27.7-1+cuda13.0_aarch64.txzz2.25.1z2.16.5z!nccl_2.25.1-1+cuda12.8_x86_64.txzz"nccl_2.25.1-1+cuda12.8_aarch64.txzz!nccl_2.16.5-1+cuda11.8_x86_64.txzz"nccl_2.16.5-1+cuda11.8_aarch64.txzr.c Csxz t||WdStjy;tdztj|ddtjdd|d|gddWYdStjy:d}t |ww) Nz`The archive format is not supported in your Python environment. Falling back to "tar" command...T)exist_oktarZxfz-C)checkz2Failed to extract the archive using "tar" command.) shutilunpack_archive ReadErrorprintosmakedirs subprocessrunCalledProcessError RuntimeError)r extract_dirmsgrrr_unpack_archivesr?c Cs~|dkr |dvr n|dkrtdd}t}||D] }|d|kr#nqtdtdd||D|dur>tjd }t|||||}tj|rTtd |t d |}|d  |d}|durptd ||dkrt dt dt n|dkrt dt dt n|dkrnJt d||||d||d} t d| t } ttj| tj| d%} tj| } | | Wdn1swYWdn1swYt dtj| d} t| j| t| }t|dksJ|d}t d|dkrEtj dkr&dd gnd g}|d!d"gD]}ttj| ||tj||q/n`|dkr|d#rX|d$krXd%}n|d&r`d'}d"}ttj| |d!tj|d!ttj| |d |tj|d ttj| |||n|dkrttj| ||nJt d(Wdn 1swYt d)dS)*Nr.)x86_64aarch64r@zK Currently this tool only supports x86_64 or aarch64 architecture for NCCL.rzB The CUDA version specified is not supported. Should be one of {}.cSsg|]}|dqS)rr)r xrrrrszinstall_lib..z~/.cupy/cuda_libzb The destination directory {} already exists. Remove the directory first if you want to reinstall.:rz, The current platform ({}) is not supported.rzsBy downloading and using cuDNN, you accept the terms and conditions of the NVIDIA cuDNN Software License Agreement:z; https://docs.nvidia.com/deeplearning/cudnn/sla/index.htmlr'zyBy downloading and using cuTENSOR, you accept the terms and conditions of the NVIDIA cuTENSOR Software License Agreement:z4 https://docs.nvidia.com/cuda/cutensor/license.htmlFz#Installing {} {} for CUDA {} to: {}rzDownloading {}...wbz Extracting...extractrz Installing...win32binlibincludeLICENSEz11.z11.0Z11z12.Z12zCleaning up...zDone!)r<library_recordsr)strr7path expandusercalculate_destinationexistsrsystemgetr6tempfileTemporaryDirectoryopenjoinbasenameurllibrequesturlopenwritereadr?namelistdirlensysr3move startswith)rprefixlibraryarchrecordZ lib_records destinationZtarget_platformZassetrZtmpdirfresponseZoutdirsubdirdir_namelibdirsitemlicenserrr install_libs                &rpcCstj||||S)zqCalculates the installation directory. ~/.cupy/cuda_lib/{cuda_version}/{library_name}/{library_version} )r7rNrW)rdrrIZlib_verrrrrPsrPcCs&t}|jdgdddd|jdtddd|jd d d gdd d |jdtddd|jdddgddd ||}|jdur_tj }|dvrPd |_n|d krXd |_nt d||j durlt j |j |_ |jdkr~t|j|j |j|jdS|jdkrttjt|jdddSJ)Nz --library)rr'r.TzLibrary to install)choicesrequiredhelpz--cudaz CUDA version)typerrrsz--archr@rAz&Target arhitecture (x86_64 or aarch64))rqdefaultrsz--prefixzInstall destination)rtrursz--actioninstalldumpzAction to perform)r@amd64zunsupported architecture: )indent)argparseArgumentParser add_argumentrM parse_argsrfrunamemachinelowerAssertionErrorrdr7rNabspathactionrprrer6jsondumpsrL)argsparserparamsrrrrmainsJ         r__main__rF)__doc__r{rr7rr3r9rarTurllib.requestrYZ_cudnn_recordsZ_cutensor_recordsZ _nccl_recordsrLrrr!appendr#r*r+r-r/r?rprPr__name__argvrrrrsf   [(