˲jaddlZddlZddlZddlZddlZddlZddlZddlmZ ddl Z dZ n#e e e ef$rdZ YnwxYwedgdZgdZgdZgd Zgd Zd Zd Zd ZdZd+dZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(d Z)d+d!Z*d"Z+d#Z,d$Z-d%Z.d&/Z0d'Z1d(Z2d)Z3e4d*kr e3dSdS),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)torchnumpytritonoptree)z cuda-cudartz cuda-cuptizcuda-librariesz cuda-openclz cuda-nvrtcz cuda-runtimecublascudnncufftcurandcusolvercusparsenccl nvjitlinknvtx) cudatoolkitsoumithmklmagma)mypyflake8onnxct|turdnd}tj|tjtj|}|\}}|j}tdkrd}ntj }| |}| |}|| | fS)z%Return (return-code, stdout, stderr).TF)stdoutstderrshellwin32oem) typestr subprocessPopenPIPE communicate returncode get_platformlocalegetpreferredencodingdecodestrip) commandr5p raw_outputraw_errrcencoutputerrs HC:\Projects\Historical\venv\Lib\site-packages\torch/utils/collect_env.pyrunrMZsMMS((DDeE *u > > >A--//J B~~  )++   s # #F ..  C v||~~syy{{ **c4||\}}}|dkrdS|S)zIRun command using run_lambda; reads and returns entire output if rc is 0.rN run_lambdarDrHout_s rLrun_and_read_allrUjs*G$$JBQ Qwwt JrNc||\}}}|dkrdStj||}|dS|dS)zIRun command using run_lambda, returns the first regex match if it exists.rN)researchgroup)rRrDregexrHrSrTmatchs rLrun_and_parse_first_matchr]rsPG$$JBQ Qwwt IeS ! !E }t ;;q>>rNcf||\}}}|dkrdS|ddS)zKRun command using run_lambda and returns first line if output is not empty.rN )splitrQs rLrun_and_return_first_linera|s:G$$JBQ Qwwt 99T??1 rNc*ttztztjdd}t |d|}||Sdfd| DS)N CONDA_EXEcondaz{} listr_c3~K|]6dtfdD2V7dS)#c3 K|]}|vV dSNrP.0namelines rL z/get_conda_packages...s'22 222222rNN) startswithanyrjrlpatternss @rLrmz%get_conda_packages..sq s## 2222222 2 2  rN) CONDA_PATTERNSCOMMON_PATTERNSNVIDIA_PATTERNSr environgetrUformatjoin splitlines)rRrqrdrSs ` rLget_conda_packagesrzs!O3oE JNN; 0 0E :y'7'7'>'> ? ?C { 99NN$$  rNc$t|ddS)Nz gcc --versionzgcc (.*)r]rRs rLget_gcc_versionr~s $Z+ N NNrNc$t|ddS)Nzclang --versionzclang version (.*)r|r}s rLget_clang_versionrs $Z1BDY Z ZZrNc$t|ddS)Nzcmake --versionz cmake (.*)r|r}s rLget_cmake_versionrs $Z1BM R RRrNctdkrd}t||dSt}t||dS)Ndarwinzkextstat | grep -i cudazcom[.]nvidia[.]CUDA [(](.*?)[)]zDriver Version: (.*?) )r?r]get_nvidia_smi)rRcmdsmis rLget_nvidia_driver_versionrsP~~!!'(S)KMM M   C $Z6O P PPrNctdks2trttjdrtjjtrtjrtjjMtjd}t|drd |j }nd}nd}tj d|zSdSt}tjd}||d z\}}}|dkrdStj|d|S) Nrhipr gcnArchNamez ({})NoGCNArchNameOnOldPyTorchz \(UUID: .+?\)z -L)r?TORCH_AVAILABLEhasattrrversionrcuda is_availableget_device_propertiesrwrget_device_namerrXcompilesub)rRpropgcnArchr uuid_regexrHrSrTs rL get_gpu_infors~~!!o!'%-QV:W:W!\a\i\m\y  >uz6688 >} ,z77::4//:%nnT-=>>GG9GG:--d33g= =t   C-..JC%K((JBQ Qwwt 6*b# & &&rNc$t|ddS)Nznvcc --versionzrelease .+ V(.*)r|r}s rLget_running_cuda_versionrs $Z1ACV W WWrNctdkrxtjdd}tjdd}tj|dd}d||}ntd krd }nd }||\}}}t|d ks |d krg|d kratjd}|>tj|rtj |SdSt} | dD]U} tj | } tj| r| | V| sdSt| } t| d kr| d Sd| } d| S)zHReturn a list of libcudnn.so; it's hard to tell which one is being used.r6 SYSTEMROOT C:\Windows CUDA_PATHz %CUDA_PATH%System32wherez{} /R "{}\bin" cudnn*.dllrz ls /usr/local/cuda/lib/libcudnn*z7ldconfig -p | grep libcudnn | rev | cut -d" " -f1 | revrrW CUDNN_LIBRARYNr_z!Probably one of the following: {})r?r rurvpathrxrwlenisfilerealpathsetr`addsorted) rR system_root cuda_path where_cmd cudnn_cmdrHrSrTl files_setfnfilesresults rLget_cudnn_versionrs~~  jnn\=AA JNN; >> GLLj'BB 077 9MM 8 # # 7 M I&&JBQ 3xx1}}qR1WW JNN? + + =RW^^A..=7##A&& &tIiioo W  b ! ! 7>>"    MM"    t 9  E 5zzQQx YYu  F / 6 6v > >>rNcd}tdkrtjdd}tjdd}tj|dd|}tj|d |}||g}|D]8}tj|rd |}n9|S) Nz nvidia-smir6rr PROGRAMFILESzC:\Program FileszNVIDIA CorporationNVSMIrz"{}")r?r rurvrrxexistsrw)rrprogram_files_root legacy_pathnew_pathsmis candidate_smis rLrrs C~~  jnn\=AA Z^^Nz get_cpu_info..Rs&#H#H#HFQqKKAKK#H#H#HrNc"g|] \}}|d| SrrPrs rLrz get_cpu_info..Ts&FFFA1  FFFrNr_rz"sysctl -n machdep.cpu.brand_stringNone) r?jsonloadsr8listappendextenditems ValueErrorr9rx) rRrHrSrKlstobjoers rL get_cpu_inforAsLBS~~  !z'** C 7 " "!z   C 77C #joo99$$ JJ #;<<< #H#Haggii#H#H#HIIIIJJJFF#))++FFFGGG # # # 3 3q66"""""""" #))C..C 8 # #!z"FGG CH Qww OsB'C>> E 7EE c"tjdrdStjdrdStjdrdStjdrdStjS)Nrr6cygwinr)sysplatformrnrPrNrLr?r?cs~ |w'' w   ) )w   * *x   * *x|rNc$t|ddS)Nzsw_vers -productVersionz(.*)r|r}s rLget_mac_versionrps $Z1JG T TTrNct|d} tj|}|dd|dd|dd}n,#t$r}|dt |z }Yd}~nd}~wwxYw|S) Nzpowershell.exe "gwmi -Class Win32_OperatingSystem | Select-Object -Property Caption, OSArchitecture,Version | ConvertTo-Json"Captionz (Version OSArchitecture)r_)rUrrrr9)rRretrrs rLget_windows_versionrts  2  C jooYLL3y>LLC8H4ILLL  }CFF}} Js1A A-A((A-c$t|ddS)Nzlsb_release -azDescription:\t(.*)r|r}s rLget_lsb_versionrs $Z1ACX Y YYrNc$t|ddS)Nzcat /etc/*-releasezPRETTY_NAME="(.*)"r|r}s rLcheck_release_filers $Z1E%: < <..s'11tt|111111rNN)rorps @rLrmz#get_pip_packages..s[  1111111 1 1 rN) PIP_PATTERNSrsrtrrr rurU executablerxry)rRrqrrS filtered_outs ` rLget_pip_packagesrs/1OCKNc11&&uK25BJ./ :IZ'[ \ \C99NN$$L  $$rNcFtjdd}|S)NPYTORCH_CUDA_ALLOC_CONFr)r rurv) ca_configs rLget_cachingallocator_configrs 8"==I rNctr^tjr@tjt jdd}|SdS)NCUDA_MODULE_LOADINGrr)rrrrinitr rurv)configs rLget_cuda_module_loading_configrsQ5:2244  5r:: urNcXtr"ddl}t|jjjSdS)Nrr)rtorch.backends.xnnpackr9backendsxnnpackenabled)rs rLrrs1%%%%5>)1222urNc t}t|\}}trtj}t tjj}t tj }tjj}ttjdrtjj dx}x}} nnd} tj d} | | d}| | d} d}tjj }ndx}x}x}}dx}x}} tjdd} t#|} t%d$id |d |d d | tjd zdt-d|d|dt/|dt1dt3|dt5|dt7|d|d|d| d|d|d| dt9|dt;dt=|dt?|d tA|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. rNrc8fd|D}|r|dndS)NcPg|]"}|v|ddd#S)NrW)rsplit)rjsprefixs rLrz;get_env_info..get_version_or_na..s1JJJ!fkkq))"-kkkrNrrrP)cfgr_lsts ` rLget_version_or_naz'get_env_info..get_version_or_nas/JJJJsJJJ"&1tAwwE1rNr_z HIP RuntimeMIOpenrrrr z{} ({}-bit runtime)rWrrrrrrrrrrrrrrr r rr r rrrrP)$rMrrr __version__r9rdebugrrrr_C _show_configr`rreplacerzrrwmaxsize bit_lengthrrrrrrrrr~rrrrr)rRrpip_list_output version_strdebug_mode_strcuda_available_strcuda_version_strrrrrr sys_versionrs rL get_env_infor&s- J#3J#?#? KT' U]011 !8!8!:!:;; =-u}e,, 5 0A0IRW W W#69O9O 2 2 2(''))//55C"3"3C"G"G %6%6sH%E%E "$ #(=#4 OTT TnT'9!>!>/ 023331 2j)))3 rNaP 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} c dd}dd}dd}dd }d }| ||j d <gd }|d gz}t fd|D}tr6tjs|r|D]} d | <|jd d<|  |  | d d<| d d< dr-| dd|j  d< dr| dd d<|j d<tjdi S)NCould not collectcP|D]}|| |||<|Srhkeys)dct replacementkeys rL replace_nonesz!pretty_str..replace_nonesHs688:: # #C3x#"CHH rNYesNocr|D]!}||dur|||<||dur|||<"|S)NTFr*)r,truefalser.s rL replace_boolsz!pretty_str..replace_boolsOsQ88:: ! !C3x4CSU"" C rN [prepend]ct|d}fd|D}d|S)Nr_cg|]}|zSrPrP)rjrltags rLrz/pretty_str..prepend..Ys666t666rN)r`rx)textr9lines updated_liness ` rLprependzpretty_str..prependWs? 4  6666666 yy'''rNNo relevant packagesc4|t|dkr|S|Sr)r)r:r-s rLreplace_if_emptyz$pretty_str..replace_if_empty\s!  D Q  rNc|;t|ddkrd|S|S)Nr_rWz {} )rr`rw)strings rLmaybe_start_on_next_linez,pretty_str..maybe_start_on_next_lineas=  #fll4&8&8"9"9A"="=??6** * rNr)rrrrc3,K|]}|duVdSrhrP)rjfield mutable_dicts rLrmzpretty_str..tsF*F*F(- Ut#*F*F*F*F*F*FrNzNo CUDArrrrz[{}] z[conda] r)r()r0r1)r6)r>rP) _asdictrallrrrrrrwrr env_info_fmt) envinfor/r5r=r@rCdynamic_cuda_fieldsall_cuda_fieldsall_dynamic_cuda_fields_missingrErFs @rL pretty_strrNGs5((((   ??$$L ! !:;;$% *_,==O&)*F*F*F*F1D*F*F*F'F'F#;uz6688;=\;$ , ,E"+L    ( 04:L0 1!=..L!=..L$4#3L4P#Q#QL %5%5lCS6T%U%UL!"N#T'.w|N/K/6~~g>Q/R/R(T(T ^$$%=)0>N1O1;*=*= %&&/L   . . . ..rNc8ttS)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. )rNr&rPrNrLget_pretty_env_inforPs lnn % %%rNctdt}t|tr=ttdr)ttjdrtjjjtj dkrtj rψfdtj D}t|tj j}tj |}t j|d}d||d z}t|tj dSdSdSdSdSdS) Nz%Collecting environment information...utils_crash_handlerrcPg|]"}tj|#SrP)r rrx)rjdump minidump_dirs rLrzmain..s)[[[$RW\\,55[[[rN)r.z%Y-%m-%d %H:%M:%Sz. *** Detected a minidump at {} created on {}, zKif this is related to your bug please include it when you file a report ***)file)printrPrrrrRrSDEFAULT_MINIDUMP_DIRrrr rrlistdirmaxgetctimedatetime fromtimestampstrftimerwr4)rJdumpslatestctime creation_timemsgrVs @rLmainresZ 1222 " "F &MMM (75'22 (wu{L\7]7] ({1F <7 " "rw~~l'C'C "[[[["*\BZBZ[[[EBG$4555FG$$V,,E$-;;EBBKKL_``MCJJ6S`aa_`C #CJ ' ' ' ' ' ' ( ( ( ( ( ( " " " "rN__main__rh)5r]rr@rXr:rr collectionsrrr ImportError NameErrorAttributeErrorOSErrorrrsrtrrrrMrUr]rarzr~rrrrrrrrr?rrrrrrrrrrrr&rCrIrNrPre__name__rPrNrLrms+   """"""LLLOOY8OOO J{%%%   8$  + + +     OOO[[[SSSQQQ''',XXX"?"?"?J   vD   UUU   ZZZ<<< < )))%%%%( FFFP< EGG= BH/H/H/V & & &((($ zDFFFFFs + 99