`i, dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z gZ gZ gZ iZdZdZdZe ede ede ed<d Zd Zd Ze ede ede ed <d ZdZe edddddde edddddde edddddde ed<dZdZdZdZedkreejdddSdS) zM CUDA Library Installer Installs the latest CUDA library supported by CuPy. Ncd|d|zS)NzAhttps://developer.download.nvidia.com/compute/cudnn/redist/cudnn//platformfilenames o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/tools/install_library.py_make_cudnn_urlr s# L   h   !"cddgd}|td|fd|Ddtd|fd|Ddd d S) N.r) _ops_infer _ops_train _cnn_infer _cnn_train _adv_infer _adv_trainz linux-x86_64c g|] }d|d S)libcudnnz.so.r).0suffixpublic_versions r z'__make_cudnn_record..2s:999"(FEE^EE999r url filenameszwindows-x86_64c"g|] }d|dd S)cudnn64_z.dllr)rr major_versions r rz'__make_cudnn_record..7s=999"(EfDDDDD999r  Linux:x86_64zWindows:x86_64)cudarassets)splitr ) cuda_versionrfilename_linuxfilename_windows suffix_listr!s ` @r __make_cudnn_recordr+$s"((--a0M///K'~~FF9999,7999 ''79IJJ9999,7999     r ct|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)intr&r+)r' cuda_majors r _make_cudnn_recordr1>si\'',,Q/00J  ! ! ! ! gE*EEEDJDDD F FFr z12.xz11.xrcdd|d|zS)Nz7https://developer.download.nvidia.com/compute/cutensor/zredist/libcutensor/z-x86_64/rrs r _make_cutensor_urlr3Ms% B:h:::: ;BB188HHI *)5EFF,-     r c(t|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:)r's r _make_cutensor_recordr<hs# "gw98 : ::r r7cdd|d|zS)Nz:https://developer.download.nvidia.com/compute/redist/nccl/vrr)rrs r _make_nccl_urlr?xs% E'N''X'' ()r c|||t||d|gdt||d|gdddS)Nz libnccl.so.{}r)r#z Linux:aarch64)r$ncclr8r%)r?r9)r' full_versionrr8filename_linux_x86_64filename_linux_aarch64s r _make_nccl_recordrEs,&"$9;;-44\BBC &"$:<<-44\BBC     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.txzrAc( tj||dS#tj$ritd t j|dt jdd|d|gdYdS#t j$rd}t|wxYwwxYw) Nz`The archive format is not supported in your Python environment. Falling back to "tar" command...T)exist_oktarxfz-C)checkz2Failed to extract the archive using "tar" command.) shutilunpack_archive ReadErrorprintosmakedirs subprocessrunCalledProcessError RuntimeError)r extract_dirmsgs r _unpack_archiverWs $h 44444   $ $ $ > ? ? ? $ K d 3 3 3 3 Nhk:$ H H H H H H H, $ $ $FCs## # $ $sB1A,,!B  Bc  |dkr|dvrn|dkrtdd}t}||D]}|d|krn@tdtd||D|tjd}t|||||}tj|r"td |tj d |}|d  |d}|"td ||d kr-tdtdtn<|dkr-tdtdtn |dkrnJtd||||d||d} td| tj5} ttj| tj| d5} t$j| 5} | | dddn #1swxYwYdddn #1swxYwYtdtj| d} t/| j| t j| }t5|dksJ|d}td|d krut6j dkrddgndg}|d d!gzD]T}t9jtj| ||tj||UnT|dkr|d"r |d#krd$}n|d%rd&}d!}t9jtj| |d tj|d t9jtj| |d|tj|dt9jtj| |||n<|dkr4t9jtj| ||nJtd'dddn #1swxYwYtd(dS))NrA)x86_64aarch64rYzK Currently this tool only supports x86_64 or aarch64 architecture for NCCL.r$zB The CUDA version specified is not supported. Should be one of {}.cg|] }|d S)r$r)rxs r rzinstall_lib..s#L#L#L!AfI#L#L#Lr z~/.cupy/cuda_libzb The destination directory {} already exists. Remove the directory first if you want to reinstall.:r%z, 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.htmlr7zyBy 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.011z12.12zCleaning up...zDone!)rTlibrary_recordsr9strrOpath expandusercalculate_destinationexistsrsystemgetrNtempfileTemporaryDirectoryopenjoinbasenameurllibrequesturlopenwritereadrWnamelistdirlensysrKmove startswith)r$prefixlibraryarchrecord lib_records destinationtarget_platformassetrtmpdirfresponseoutdirsubdirdir_namelibdirsitemlicenses r install_librs?&T%:::   NOO O F!Kg&PP &>T ! ! E "s#L#L{77K#L#L#LMMNNPP P~##$677'gvgOOK w~~k""M88>{8K8KMM M"**33T33O 8  $ 7 7E }//5vo/F/FHH H' L M M M KLLL  J   O P P P DEEE  F   u / 6 6&.+ ? ?@@@ ,C  $ $S ) )***  $ & &% & "',,vrw'7'7'<'<==t D D )'',, ) ((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ofi00'''F##6{{a!9 o g  (+ (?(?uenneWG9i"88 5 5 GLL488GLLd335555 5 " "u%% $&..'' G K VXy99 [)44 6 6 6 K VXud;; [%00 2 2 2 K VXw77 F F F F    K VX66 D D D D 5 K% % % % % % % % % % % % % % % L 'NNNNNs\1A U!= K((K K(K K(K K( U!(K, ,U!/K, 0I%U!!U%(U%cFtj||||S)zqCalculates the installation directory. ~/.cupy/cuda_lib/{cuda_version}/{library_name}/{library_version} )rOrjrs)rr$rclib_vers r rlrls 7<<c7 3 33r cptj}|dgddd|dtdd|d d d gdd |dtdd|dddgdd ||}|jVt jj }|dvrd |_n |d krd |_ntd||j )tj |j |_ |jdkr(t!|j|j |j|jdS|jdkr5t't)jt,|jddSJ)Nz --library)rr7rATzLibrary to install)choicesrequiredhelpz--cudaz CUDA version)typerrz--archrYrZz&Target arhitecture (x86_64 or aarch64))rdefaultrz--prefixzInstall destination)rrrz--actioninstalldumpzAction to perform)rYamd64zunsupported architecture: )indent)argparseArgumentParser add_argumentri parse_argsrrunamemachinelowerAssertionErrorrrOrjabspathactionrr$rrNjsondumpsrh)argsparserparamsrs r mainrs  $ & &F  = = =!%1333 sT+--- 8Y*? $EGGG  d2444  Y,? )0222  t $ $F {.""*0022 ) ) )"FKK  ! !#FKK !Gg!G!GHH H }  66  } !!FKK ! ! ! ! ! &  dj8CCCDDDDDur __main__r`)__doc__rrrOrrKrQr}rpurllib.requestru_cudnn_records_cutensor_records _nccl_recordsrhr r+r1appendr3r:r<r?rErWrrlr__name__argvrr r rs    """4FFF((00111((00111)<<<(:::..v66777..v66777/ ))).&& Hh'(**+++&& Hh'(**+++&& Hh'(**+++( $ $ $XXXv444%%%P zD!""r