DŽg=[ddlZddlZddlZddlZddlZddlZddlmZ ddlZdZ edgdZhdZhdZd Zd Zd Zd Zd)d ZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&d)dZ'd Z(d!Z)d"Z*d#Z+d$jYZ-d%Z.d&Z/d'Z0e1d(k(re0yy#e e e e f$rdZ YwxYw)*N) namedtupleTF SystemEnv) torch_versionis_debug_buildcuda_compiled_version gcc_version clang_version cmake_versionos libc_versionpython_versionpython_platformis_cuda_availablecuda_runtime_versioncuda_module_loadingnvidia_driver_versionnvidia_gpu_models cudnn_version pip_version pip_packagesconda_packageship_compiled_versionhip_runtime_versionmiopen_runtime_versioncaching_allocator_configis_xnnpack_availablecpu_info>mklmagmanumpytorchoptreetritonsoumith cudatoolkit>mypyonnxr r!flake8r"r#ct|turdnd}tj|tjtj|}|j \}}|j }tdk(rd}ntj}|j|}|j|}||j|jfS)z%Return (return-code, stdout, stderr).TF)stdoutstderrshellwin32oem) typestr subprocessPopenPIPE communicate returncode get_platformlocalegetpreferredencodingdecodestrip) commandr,p raw_outputraw_errrcencoutputerrs _/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/utils/collect_env.pyrunrDHsMS(DeE *u >A--/J B~ ))+   s #F .. C v||~syy{ **c*||\}}}|dk7ry|S)zIRun command using run_lambda; reads and returns entire output if rc is 0.rN run_lambdar;r?out_s rCrun_and_read_allrLXs!G$JBQ Qw JrEcz||\}}}|dk7rytj||}|y|jdS)zIRun command using run_lambda, returns the first regex match if it exists.rN)researchgroup)rIr;regexr?rJrKmatchs rCrun_and_parse_first_matchrT`sBG$JBQ Qw IIeS !E } ;;q>rEcN||\}}}|dk7ry|jddS)zKRun command using run_lambda and returns first line if output is not empty.rN )splitrHs rCrun_and_return_first_linerXjs/G$JBQ Qw 99T?1 rEcttjjdd}t |dj |}||Sdj fd|jDS)N CONDA_EXEcondaz{} listrVc3nK|]+jdstfdDr-yw)#c3&K|]}|v ywNrG.0namelines rC z/get_conda_packages...~s2 2N) startswithanyrarcpatternss @rCrdz%get_conda_packages..zs3 s# 22 2 s15)DEFAULT_CONDA_PATTERNSr environgetrLformatjoin splitlines)rIrir[rJs ` rCget_conda_packagesrprsh) JJNN; 0E :y'7'7'> ?C { 99NN$ rEct|ddS)Nz gcc --versionzgcc (.*)rTrIs rCget_gcc_versionrts $Z+ NNrEct|ddS)Nzclang --versionzclang version (.*)rrrss rCget_clang_versionrvs $Z1BDY ZZrEct|ddS)Nzcmake --versionz cmake (.*)rrrss rCget_cmake_versionrxs $Z1BM RRrEchtdk(rd}t||dSt}t||dS)Ndarwinzkextstat | grep -i cudazcom[.]nvidia[.]CUDA [(](.*?)[)]zDriver Version: (.*?) )r6rTget_nvidia_smi)rIcmdsmis rCget_nvidia_driver_versionr~s?~!'(S)KM M  C $Z6O PPrEctdk(s:trttjdrtjj trtj jrtjj Jtj jd}t|drdj|j}nd}nd}tj jd|zSyt}tjd}||d z\}}}|dk7rytj|d|S) Nrzhipr gcnArchNamez ({})NoGCNArchNameOnOldPyTorchz \(UUID: .+?\)z -L)r6TORCH_AVAILABLEhasattrr!versionrcuda is_availableget_device_propertiesrmrget_device_namer{rOcompilesub)rIpropgcnArchr} uuid_regexr?rJrKs rC get_gpu_infors~!o'%--QV:W\a\i\i\m\m\y uzz668}}  ,zz77:4/%nnT-=-=>G9G::--d3g= =  C-.JC%K(JBQ Qw 66*b# &&rEct|ddS)Nznvcc --versionzrelease .+ V(.*)rrrss rCget_running_cuda_versionrs $Z1ACV WWrEctdk(rttjjdd}tjjdd}tjj |dd}dj ||}ntd k(rd }nd }||\}}}t|d k(s |d k7re|d k7r`tjjd}|>tjj|rtjj|Syt} |jdD]R} tjj| } tjj| sB| j| T| syt| } t| d k(r| d Sdj | } dj | S)zHReturn a list of libcudnn.so; it's hard to tell which one is being used.r- SYSTEMROOT C:\Windows CUDA_PATHz %CUDA_PATH%System32wherez{} /R "{}\bin" cudnn*.dllrzz ls /usr/local/cuda/lib/libcudnn*z7ldconfig -p | grep libcudnn | rev | cut -d" " -f1 | revrrN CUDNN_LIBRARYNrVz!Probably one of the following: {})r6r rkrlpathrnrmlenisfilerealpathsetrWaddsorted) rI system_root cuda_path where_cmd cudnn_cmdr?rJrKl files_setfnfilesresults rCget_cudnn_versionrsj~ jjnn\=A JJNN; > GGLLj'B 077 9M 8 # 7 M I&JBQ 3x1}qR1W JJNN? + =RWW^^A.77##A& &Iiio WW  b ! 77>>"  MM"   9 E 5zQQx YYu F / 6 6v >>rEcd}tdk(rtjjdd}tjjdd}tjj |dd|}tjj |d |}||g}|D]5}tjj |s#d j|}|S|S) Nz nvidia-smir-rr PROGRAMFILESzC:\Program FileszNVIDIA CorporationNVSMIrz"{}")r6r rkrlrrnexistsrm)r}rprogram_files_root legacy_pathnew_pathsmis candidate_smis rCr{r{s C~ jjnn\=A ZZ^^N.run_with_pip...s5D44<5reN)rgrhs @rCrdz9get_pip_packages..run_with_pip..s' 5H55  s $)rLrnro)piprJrirIs rC run_with_pipz&get_pip_packages..run_with_pips?z3&:K1L+LMyy (   rEr3pip3rz-mpip)DEFAULT_PIP_PATTERNSrr executable)rIrirrrJs`` rCget_pip_packagesrsI'  KKNc1&uK 0 1C  rEcFtjjdd}|S)NPYTORCH_CUDA_ALLOC_CONFr)r rkrl) ca_configs rCget_cachingallocator_configrs 8"=I rEctr^tjjr@tjj t j jdd}|Sy)NCUDA_MODULE_LOADINGrr)rr!rrinitr rkrl)configs rCget_cuda_module_loading_configrs?5::224  5r: rEcjtr-ddl}t|jjj Sy)Nrr)rtorch.backends.xnnpackr0backendsxnnpackenabled)r!s rCrrs&%5>>))1122rEc t}t|\}}trtj}t tj j}t tjj}tj j}ttj drtj jdx}x}} nmd} tjjjd} | | d}| | d} d}tj j}ndx}x}x}}dx}x}} tj j!dd} t#|} t%d#id|d |d d j'| tj(j+d zd t-d|d|dt/|dt1dt3|dt5|dt7|d|d|d| d|d|d| dt9|dt;dt=|dt?|dtA|d tCd!tEd"tG|S)$a Collects environment information to aid in debugging. The returned environment information contains details on torch version, is debug build or not, cuda compiled version, gcc version, clang version, cmake version, operating system, libc version, python version, python platform, CUDA availability, CUDA runtime version, CUDA module loading config, GPU model and configuration, Nvidia driver version, cuDNN version, pip version and versions of relevant pip and conda packages, HIP runtime version, MIOpen runtime version, Caching allocator config, XNNPACK availability and CPU information. Returns: SystemEnv (namedtuple): A tuple containining various environment details and system information. rrcl|Dcgc]}||vs|jddd}}|r|dSdScc}w)NrNrr)rsplit)cfgprefixs_lsts rCget_version_or_naz'get_env_info..get_version_or_nasB7:J!fkq)"-JJ"&tAw1E1Ks 11rVz HIP RuntimeMIOpen rrr z{} ({}-bit runtime)rNrrrrrrrrrrrrrrr r rr r rrrrG)$rDrrr! __version__r0rdebugrrrr_C _show_configrWrreplacerprrmmaxsize bit_lengthrrrrr~rrrrtrvrxrrr)rIrpip_list_output version_strdebug_mode_strcuda_available_strcuda_version_strrrrrr sys_versionrs rC get_env_infor s J#3J#? K'' U]]001 !8!8!:; ==--u}}e, 0A0A0IRW W W#69O 2(('')//5C"3C"G %6sH%E "$ #(==#4#4 OTT TnT'9/ 0231 2j)3 rEaP PyTorch version: {torch_version} Is debug build: {is_debug_build} CUDA used to build PyTorch: {cuda_compiled_version} ROCM used to build PyTorch: {hip_compiled_version} OS: {os} GCC version: {gcc_version} Clang version: {clang_version} CMake version: {cmake_version} Libc version: {libc_version} Python version: {python_version} Python platform: {python_platform} Is CUDA available: {is_cuda_available} CUDA runtime version: {cuda_runtime_version} CUDA_MODULE_LOADING set to: {cuda_module_loading} GPU models and configuration: {nvidia_gpu_models} Nvidia driver version: {nvidia_driver_version} cuDNN version: {cudnn_version} HIP runtime version: {hip_runtime_version} MIOpen runtime version: {miopen_runtime_version} Is XNNPACK available: {is_xnnpack_available} CPU: {cpu_info} Versions of relevant libraries: {pip_packages} {conda_packages} cX dd}dd}dd}dd}d}|j ||j d<gd}|dgz}t fd |D}tr=tj j s|r|D]} d | < |jd d <|  |  | d  d <| d d< d r(| d dj|j d < dr| dd d<|j d<tjdi S)NcF|jD] }|| |||<|Sr_keys)dct replacementkeys rC replace_nonesz!pretty_str..replace_noness488: #C3x#"CH # rEcd|jD]}||dur|||<||dus|||<|S)NTFr )rtruefalsers rC replace_boolsz!pretty_str..replace_bools&sH88: !C3x4CSU" C  !  rEct|jd}|Dcgc]}||z }}dj|Scc}w)NrV)rWrn)texttaglinesrc updated_liness rCprependzpretty_str..prepend.s; 4 056t6 6yy''7s 5c*|t|dk(r|S|Sr)r)rrs rCreplace_if_emptyz$pretty_str..replace_if_empty3s  D Q  rEcf|.t|jddkDrdj|S|S)NrVrNz {} )rrWrm)strings rCmaybe_start_on_next_linez,pretty_str..maybe_start_on_next_line8s1  #fll4&8"9A"=??6* * rEr)rrrrc3,K|] }|du ywr_rG)rafield mutable_dicts rCrdzpretty_str..Ks"*F(- Ut#*FszNo CUDArrrrz[{}] z[conda] r)zCould not collect)YesNo)z [prepend])zNo relevant packagesrG) _asdictrallrr!rrrrmrr env_info_fmt) envinforrrrr dynamic_cuda_fieldsall_cuda_fieldsall_dynamic_cuda_fields_missingr"r#s @rC pretty_strr-s{(   ??$L !!:!:;$% *_,==O&)*F1D*F'F#uzz668=\$ ,E"+L  ,  ( ( 04:L0 1!.L!.L$4L4P#QL %5lCS6T%UL!"N#'.|N/K/6~~g>Q>Q/R(T ^$$%)0>N1O1;*= %&&//L    . ..rEc(ttS)a Returns a pretty string of environment information. This function retrieves environment information by calling the `get_env_info` function and then formats the information into a human-readable string. The retrieved environment information is listed in the document of `get_env_info`. This function is used in `python collect_env.py` that should be executed when reporting a bug. Returns: str: A pretty string of the environment information. )r-r rGrErCget_pretty_env_infor/is ln %%rEc0tdt}t|trjttdrXttj drrKs8   "O {%  8  +  O[SQ',X"?J v" UkZ< < )* FP< EG= BH/V &($ zFq Y8OsB33 CC