QVjiwddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z  ddl Z dZ n#eeeef$rdZ YnwxYwedgdZgdZgd Zgd Zgd Zgd Zd ZdZdZdZd2dZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*d Z+d!Z,d"Z-d#Z.d$Z/d%Z0d&Z1d'Z2d2d(Z3d)Z4d*Z5d+Z6d,Z7d-8Z9d.Z:d/Z;d0Z>rTcf||\}}}|dkrdS|ddS)zKRun command using run_lambda and returns first line if output is not empty.rN )splitrWs rRrun_and_return_first_linergs:G$$JBQ Qwwt 99T??1 rTc:ttztztztjdd}t|d|}||Sd fd| DS)N CONDA_EXEcondaz{} listrec3~K|]6dtfdD2V7dS)#c3 K|]}|vV dSNrV.0namelines rR z/get_conda_packages...s'+N+NTDDL+N+N+N+N+N+NrTN) startswithanyrprrpatternss @rRrsz%get_conda_packages..sq s##),+N+N+N+NX+N+N+N(N(N rT) CONDA_PATTERNSCOMMON_PATTERNSNVIDIA_PATTERNSONEAPI_PATTERNSr environgetr[formatjoin splitlines)rXrwrjrYs ` rRget_conda_packagesrs!O3oEW JNN; 0 0E :y'7'7'>'> ? ?C { 99NN$$  rTc$t|ddS)Nz gcc --versionzgcc (.*)rcrXs rRget_gcc_versionrs $Z+ N NNrTc$t|ddS)Nzclang --versionzclang version (.*)rrs rRget_clang_versionrs $%'<  rTc$t|ddS)Nzcmake --versionz cmake (.*)rrs rRget_cmake_versionrs $Z1BM R RRrTctdkrd}t||dSt}t||dS)Ndarwinzkextstat | grep -i cudazcom[.]nvidia[.]CUDA [(](.*?)[)]zDriver Version: (.*?) )rErcget_nvidia_smi)rXcmdsmis rRget_nvidia_driver_versionrsP~~!!'( ?      C $Z6O P PPrTctdks2trttjdrtjjtrtjrtjjMtjd}t|drd |j }nd}nd}tj d|zSdSt}tjd}||d z\}}}|dkrdStj|d|S) Nrhipr gcnArchNamez ({})NoGCNArchNameOnOldPyTorchz \(UUID: .+?\)z -L)rETORCH_AVAILABLEhasattrr versionrcuda is_availableget_device_propertiesr~rget_device_namerr^compilesub)rXpropgcnArchr uuid_regexrNrYrZs rR get_gpu_infors!~~!!" EM5 ) )" M  )  >uz6688 >} ,z77::4//:%nnT-=>>GG9GG:--d33g= =t   C-..JC%K((JBQ Qwwt 6*b# & &&rTc$t|ddS)Nznvcc --versionzrelease .+ V(.*)rrs rRget_running_cuda_versionrs $Z1ACV W WWrTctdkrxtjdd}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.r< 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 | revrr] CUDNN_LIBRARYNrez!Probably one of the following: {})rEr r|r}pathrr~lenisfilerealpathsetrfaddsorted) rX system_root cuda_path where_cmd cudnn_cmdrNrYrZl files_setfnfilesresults rRget_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 > >>rTcd}tdkrtjdd}tjdd}tj|dd|}tj|d |}||g}|D]8}tj|rd |}n9|S) Nz nvidia-smir<rr PROGRAMFILESzC:\Program FileszNVIDIA CorporationNVSMIrz"{}")rEr r|r}rrexistsr~)rrprogram_files_root legacy_pathnew_pathsmis candidate_smis rRrrs C~~  jnn\=AA Z^^Nintel-opencl-icdlibze1 level-zeroz intel-openclrrr* z: r<cygwinzpowershell.exe "gwmi -Class Win32_PnpSignedDriver | where{$_.DeviceClass -eq \"DISPLAY\" -and $_.Manufacturer -match \"Intel\"} | Select-Object -Property DeviceName,DriverVersion,DriverDate | ConvertTo-Json" DeviceName: DriverVersion ( DriverDate)re) rEr}rrappendr[jsonloadsr>list ValueErrorr?r) rXrplatformpkgspkgvertxtobjoes rRget_intel_gpu_driver_versionrNs C~~H7     $ #'))2 . .% & / /C' C88Ce|| -----...&&&     *S//CCyyD  AJJXQ|_XX/0BXXa oXXX  LO 4LLL8ILLLMMM    JJsOOO JJs1vv          99S>>sB D E&%7E!!E&c$g}t}|dkrt|d}|r tj|}|dg}t |t r"|r |d|Dn|dnj#ttf$rA}|||t|Yd}~nd}~wwxYw|d|dvrt|d}|r tj|}t |t r"|r |d|Dn,|d |d dnc#t$rA}|||t|Yd}~nd}~wwxYw|dd |S) Nrzxpu-smi discovery -j device_listc3,K|]}d|dVdS)r device_nameNrVrpdevices rRrsz(get_intel_gpu_onboard..s0VV;F=$9;;VVVVVVrTrrzpowershell.exe "gwmi -Class Win32_PnpSignedDriver | where{$_.DeviceClass -eq \"DISPLAY\" -and $_.Manufacturer -match \"Intel\"} | Select-Object -Property DeviceName | ConvertTo-Json"c3,K|]}d|dVdS)rrNrVrs rRrsz(get_intel_gpu_onboard..s0MMv:F<$8::MMMMMMrTrrre) rEr[rrr} isinstancerextendrr TypeErrorr?r)rXrrrrrrs rRget_intel_gpu_onboardrs8C~~H7z+ABB   #joo!ggmR88 k400&[&JJVV+VVVVVVVJJu%%% * # # # 3 3q66"""""""" # JJu   &&&  m     #jooc4((DSDJJMMMMMMMMMJJBCGGL%$@$@BBCCC # # # 3 3q66"""""""" # JJu    99S>>s1A6B!!C327C..C3"A7F G%$7G  G%ctrttdsdStj}|dkrdSdt |D}d|S)NxpurrcXg|]'}d|dtj|(S)z* [z] )r rr)rpis rR z*get_intel_gpu_detected..sF<=7a77592215577rTre)rrr r device_countranger)rXrdevicess rRget_intel_gpu_detectedr s} '%"7"7u9))++LquAF|ATATG 99W  rTc$d\}}}tdkr|d\}}}nYtdkr%|d\}}}|dkrg} tj|}t|turL|D]H}|d|d|DIn1|d |DnN#t$rA}|||t|Yd}~nd}~wwxYwd |}n!td kr|d \}}}d }|dkr|}n|}|S)N)rrrrlscpur<zpowershell.exe "gwmi -Class Win32_Processor | Select-Object -Property Name,Manufacturer,Family, Architecture,ProcessorType,DeviceID,CurrentClockSpeed,MaxClockSpeed,L2CacheSize,L2CacheSpeed,Revision | ConvertTo-Json"rz----------------------c"g|] \}}|d| SrrVrpkvs rRrz get_cpu_info..s&#H#H#HFQqKKAKK#H#H#HrTc"g|] \}}|d| Sr rVrs rRrz get_cpu_info..s&FFFA1  FFFrTrerz"sysctl -n machdep.cpu.brand_stringNone) rErrr>rrritemsrr?r) rXrNrYrQrrrrrs rR get_cpu_inforsLBS~~  !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)Nrr<rr)sysrrtrVrTrRrErE"s~ |w'' w   ) )w   * *x   * *x|rTc$t|ddS)Nzsw_vers -productVersionz(.*)rrs rRget_mac_versionr/s $Z1JG T TTrTct|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"CaptionrVersionrOSArchitecturerre)r[rrrr?)rXrrrs rRget_windows_versionr3s  2  C jooYLL3y>LLC8H4ILLL  }CFF}} Js1A A-A((A-c$t|ddS)Nzlsb_release -azDescription:\t(.*)rrs rRget_lsb_versionrAs $$&;  rTc$t|ddS)Nzcat /etc/*-releasezPRETTY_NAME="(.*)"rrs rRcheck_release_filer!Gs $(*?  rTcddlm}t}|dvrt|S|dkr1t |}|dSd||S|dkr|t |}|d||St|}|d||Sd||S|S)Nr)machinerrz macOS {} ({})rz{} ({}))rr#rErrr~rr!)rXr#rrdescs rRget_osr%Ms      ~~H&&&":...8!*-- ?4%%gwwyy9997z**  ##D''))44 4"*--  ##D''))44 4''))444 OrTc2ddl}|SNrrr(s rRget_python_platformr)lsOOO     rTcddl}tdkrdSd|S)Nrrr-)rrErlibc_verr(s rRget_libc_versionr-rs=OOO~~  u 88H%%'' ( ((rTcLttztztztjjdkrdnd}dtjd<t|tj ddd g}|||fSd fd | D}||fS) zZReturn `pip list` output. Note: will also find conda-installed pytorch and numpy packages.Npip3pip1PIP_DISABLE_PIP_VERSION_CHECKz-mpiprz--format=freezerec3TK|]!tfdDV"dS)c3 K|]}|vV dSrnrVros rRrsz-get_pip_packages...s'0S0S$0S0S0S0S0S0SrTN)rurvs @rRrsz#get_pip_packages..sZS0S0S0S0S(0S0S0S-S-S rT) PIP_PATTERNSryrzr{r version_infomajorr r|r[ executablerr)rXrwrrY filtered_outs ` rRget_pip_packagesr;zs/1OCoU,2a77&&UK25BJ./ S^Wf6GH  C {C99))L  $$rTctjdd}|s tjdd}|S)NPYTORCH_CUDA_ALLOC_CONFrPYTORCH_HIP_ALLOC_CONF)r r|r}) ca_configs rRget_cachingallocator_configr@s= 8"==I AJNN#;R@@ rTctr^tjr@tjt jdd}|SdS)NCUDA_MODULE_LOADINGrr)rr rrinitr r|r})configs rRget_cuda_module_loading_configrEsQ5:2244  5r:: urTcXtr"ddl}t|jjjSdS)Nrr)rtorch.backends.xnnpackr?backendsxnnpackenabled)r s rRrrs1%%%%5>)1222urTc ht}t|\}}trtj}t tjj}t tj }tjj}t tj }tj rV|ddtjj dzdt|dzdt|dzdt|z}ttjdrtjjdx}x} } npd } tjd} | | d } | | d } d}tjj}ndx}x}x}x}}dx}x} } t&jdd } t+|}t-d)id |d|dd| t&jdzdt5d|d|dt7|dt9dt;|dt=|dt?|d|d|d| d| d|d|d |d!tA|d"tCd#tE|d$tG|d%tI|d&tKd'tMd(tO|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 containing various environment details and system information. rezXPU used to build PyTorch: zIntel GPU driver version: zIntel GPU models onboard: zIntel GPU models detected: rNrc8fd|D}|r|dndS)NcPg|]"}|v|ddd#S)Nr])rsplit)rpsprefixs rRrz;get_env_info..get_version_or_na..s1JJJ!fkkq))"-kkkrTrrrV)cfgrQ_lsts ` rRget_version_or_naz'get_env_info..get_version_or_nas/JJJJsJJJ"&1tAwwE1rTz HIP RuntimeMIOpenrrrrz{} ({}-bit runtime)r]rrrrrrrrrrrrrrrr r r r r rrrrV)(rSr;rr __version__r?rdebugrrrrrr rr_C _show_configrfrreplacerrr~maxsize bit_lengthr)rrErrrr%r-rrrr@rr)rXrpip_list_output version_strdebug_mode_strcuda_available_strcuda_version_strxpu_available_strrrrrTrR sys_versionrs rR get_env_infords J#3J#?#? KT' U]011 !8!8!:!:;; =- 6 6 8 899 9 ! ! # # $(((E 0AEEEF\0LZ0X0X\\\]V0Ej0Q0QUUUVV1G 1S1SUU V  u-- 5161B1JRW W W#69O9O 2 2 2(''))//55C"3"3C"G"G %6%6sH%E%E "$ #(=#4 chh hnh'9h!>!>5 623337 8j)))9 rTau 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} Is XPU available: {is_xpu_available} 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]}|| |||<|Srnkeys)dct replacementkeys rR replace_nonesz!pretty_str..replace_nones%s688:: # #C3x#"CHH rTYesNocr|D]!}||dur|||<||dur|||<"|S)NTFrh)rjtruefalserls rR replace_boolsz!pretty_str..replace_bools,sQ88:: ! !C3x4CSU"" C rT [prepend]ct|d}fd|D}d|S)Nrecg|]}|zSrVrV)rprrtags rRrz/pretty_str..prepend..6s666t666rT)rfr)textrwlines updated_liness ` rRprependzpretty_str..prepend4s? 4  6666666 yy'''rTNo relevant packagesc4|t|dkr|S|Sr')r)rxrks rRreplace_if_emptyz$pretty_str..replace_if_empty9s!  D Q  rTc|;t|ddkrd|S|S)Nrer]z {} )rrfr~)strings rRmaybe_start_on_next_linez,pretty_str..maybe_start_on_next_line>s=  #fll4&8&8"9"9A"="=??6** * rTr)rrrrc3,K|]}|duVdSrnrV)rpfield mutable_dicts rRrszpretty_str..Rs>**(- Ut#******rTzNo CUDArrrrz[{}] z[conda] r)rf)rnro)rt)r|rV) _asdictrallrr rrrr~rr env_info_fmt) envinformrsr{r~rdynamic_cuda_fieldsall_cuda_fieldsall_dynamic_cuda_fields_missingrrs @rR pretty_strr$s7((((   ??$$L)A(@!))L$%  *_,==O&)****1D***''# ; '')); ,; % , ,E"+L    ( 04:L0 1!=..L!=..L$4#3L4P#Q#QL %5%5lCS6T%U%UL!"N# '.w  ('..9L*M*M( ( ^$$% )0 ) *J* * %& '/L   . . . ..rTc8ttS)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. )rrdrVrTrRget_pretty_env_inforws lnn % %%rTctdt}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|#SrV)r rr)rpdump minidump_dirs rRrzmain..s659 \400rT)rlz%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)printrrrr rrDEFAULT_MINIDUMP_DIRrrr rrlistdirmaxgetctimedatetime fromtimestampstrftimer~r:)rPdumpslatestctime creation_timemsgrs @rRmainrs 1222 " "F &MMM ( E7 # #( EK!1 2 2( {1F <7 " "rw~~l'C'C "=?Z =U=UEBG$4555FG$$V,,E$-;;EBBKK#MBHHM``  #CJ ' ' ' ' ' '+(((((( # " " "rT__main__rn)>rrrFr r^r@r collectionsrtypingrrr r ImportError NameErrorAttributeErrorOSErrorrryrzr{rxr6rSr[rcrgrrrrrrrrrrrrrr rrErrrr!r%r)r-r;r@rErrdrIrrrr__name__rVrTrRrs   """""" LLLOOY8OOO J   @$0 +++"    OOO SSSQQQ'''4XXX"?"?"?J$!!!H...b$$$N   tD   UUU     > )))%%%%.UUUp> EGG? DP/P/P/f & & &(((: zDFFFFFs 1 ??