o <&iW @sddlZddlZddlZddlZddlZddlZddlmZzddlZdZ Wne e e e fy5dZ YnwedgdZhdZhdZd d Zd d Zd dZddZdGddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Z d)d*Z!d+d,Z"d-d.Z#d/d0Z$d1d2Z%d3d4Z&dGd5d6Z'd7d8Z(d9d:Z)d;d<Z*d=d>Z+d?,Z-d@dAZ.dBdCZ/dDdEZ0e1dFkre0dSdS)HN) 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>optreetritontorchZsoumithZ cudatoolkitZmklZmagmanumpy>rZonnxrrZmypyr Zflake8c Cszt|turdnd}tj|tjtj|d}|\}}|j}tdkr&d}nt }| |}| |}|| | fS)z%Return (return-code, stdout, stderr).TF)stdoutstderrshellwin32Zoem) typestr subprocessPopenPIPE communicate returncode get_platformlocalegetpreferredencodingdecodestrip) commandr#p raw_outputZraw_errrcencoutputerrr8BC:\wamp64\www\opt\env\Lib\site-packages\torch/utils/collect_env.pyrunGs     r:cCs||\}}}|dkr dS|S)zIRun command using run_lambda; reads and returns entire output if rc is 0.rNr8 run_lambdar1r4out_r8r8r9run_and_read_allWsr?cCs<||\}}}|dkr dSt||}|durdS|dS)zIRun command using run_lambda, returns the first regex match if it exists.rN)researchgroup)r<r1regexr4r=r>matchr8r8r9run_and_parse_first_match_s  rFcCs(||\}}}|dkr dS|ddS)zKRun command using run_lambda and returns first line if output is not empty.rN )splitr;r8r8r9run_and_return_first_lineisrIcsRdurttjdd}t|d|}|dur|Sdfdd|DS)NZ CONDA_EXEcondaz{} listrGc3s4|]dstfddDrVqdS)#c3|]}|vVqdSNr8.0nameliner8r9 }z/get_conda_packages...N) startswithanyrOpatternsrQr9rSys z%get_conda_packages..)DEFAULT_CONDA_PATTERNSr environgetr?formatjoin splitlines)r<rYrJr=r8rXr9get_conda_packagesqsr`cC t|ddS)Nz gcc --versionzgcc (.*)rFr<r8r8r9get_gcc_version rdcCra)Nzclang --versionzclang version (.*)rbrcr8r8r9get_clang_versionrerfcCra)Nzcmake --versionz cmake (.*)rbrcr8r8r9get_cmake_versionrergcCs,tdkr d}t||dSt}t||dS)Ndarwinzkextstat | grep -i cudazcom[.]nvidia[.]CUDA [(](.*?)[)]zDriver Version: (.*?) )r,rFget_nvidia_smi)r<cmdsmir8r8r9get_nvidia_driver_versions  rlcCstdkstr5ttjdr5tjjdur5tr3tjr3tjdtjjdur0d tj dj SdSdSt }t d}||d\}}}|dkrLdSt |d|S)Nrhhipz ({})rz \(UUID: .+?\)z -L)r,TORCH_AVAILABLEhasattrrversionrmcuda is_availableZget_device_namer]Zget_device_propertiesZ gcnArchNamerirAcompilesub)r<rkZ uuid_regexr4r=r>r8r8r9 get_gpu_infos&  rvcCra)Nznvcc --versionzrelease .+ V(.*)rbrcr8r8r9get_running_cuda_versionrerwc Cs"tdkr"tjdd}tjdd}tj|dd}d||}n td kr*d }nd }||\}}}t|d ksA|d krY|d krYtjd}|durWtj|rWtj |SdSt } | dD]} tj | } tj| rt| | qa| sydSt | } 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:\WindowsZ CUDA_PATHz %CUDA_PATH%System32wherez{} /R "{}\bin" cudnn*.dllrhz ls /usr/local/cuda/lib/libcudnn*z7ldconfig -p | grep libcudnn | rev | cut -d" " -f1 | revrr@Z CUDNN_LIBRARYNrGz!Probably one of the following: {})r,r r[r\pathr^r]lenisfilerealpathsetrHaddsorted) r< system_rootZ cuda_pathZ where_cmdZ cudnn_cmdr4r=r>lZ files_setfnfilesresultr8r8r9get_cudnn_versions6          rcCs~d}tdkr=tjdd}tjdd}tj|dd|}tj|d |}||g}|D]}tj|r.run_with_pip...N)rVrWrXrQr9rSs z9get_pip_packages..run_with_pip..)r?r^r_)pipr=rYr<r8r9 run_with_pipsz&get_pip_packages..run_with_pipr3Zpip3rz-mpip)DEFAULT_PIP_PATTERNSrrq executable)r<rYrrr=r8rr9get_pip_packagess rcCstjdd}|S)NZPYTORCH_CUDA_ALLOC_CONFrn)r r[r\)Z ca_configr8r8r9get_cachingallocator_configsrcCs.trtjrtjtjdd}|SdS)NZCUDA_MODULE_LOADINGrnr)rorrrrsinitr r[r\)configr8r8r9get_cuda_module_loading_configs  rcCstr ddl}t|jjjSdS)Nrr)roZtorch.backends.xnnpackr&backendsZxnnpackenabled)rr8r8r9rsrc Cst}t|\}}trNtj}ttjj}ttj }tjj}t tjdr*tjj dur1d}}} n+dd} tj d} | | d}| | d} d}tjj }nd}}}}d}}} tjdd} t|} td$id |d |d d | tjd dtd|d|dt|dtdt|dt|dt|d|d|d| d|d|d| dt|dtdt|dt|d t |d!t!d"t"d#t#|S)%Nrmrcs"fdd|D}|r|dSdS)Ncs$g|]}|vr|dddqS)Nr@)rsplit)rOsprefixr8r9 s$z;get_env_info..get_version_or_na..rrr8)cfgrZ_lstr8rr9get_version_or_nasz'get_env_info..get_version_or_narGz HIP RuntimeZMIOpen rrr z{} ({}-bit runtime)r@r rrrrrrrrrrrrrr r rrr rrrr8)$r:rror __version__r&rqdebugrrrsrprmZ_CZ _show_configrHrreplacer`rr]maxsize bit_lengthrrwrrvrlrrrrdrfrgrrr)r<rZpip_list_output version_strZdebug_mode_strZcuda_available_strZcuda_version_strrrrrr sys_versionrr8r8r9 get_env_infos           raP 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 s*ddd}ddd}dd d }d d d }dd}|||jd<gd}|dg}tfdd|D}trRtjsR|rR|D]} d| <qB|jdurRdd<|||dd<|dd<dr{|dd|j d<dr|ddd<|j d<t jd!iS)"NCould not collectcSs(|D] }||dur q|||<q|SrMkeys)dct replacementkeyr8r8r9 replace_noness   z!pretty_str..replace_nonesYesNocSs<|D]}||dur|||<q||dur|||<q|S)NTFr)rtruefalserr8r8r9 replace_boolss    z!pretty_str..replace_bools [prepend]cs&|d}fdd|D}d|S)NrGcsg|]}|qSr8r8)rOrRtagr8r9rsz/pretty_str..prepend..)rHr^)textrlinesZ updated_linesr8rr9prepends  zpretty_str..prependNo relevant packagescSs|dur t|dkr |S|Sr)r})rrr8r8r9replace_if_emptysz$pretty_str..replace_if_emptycSs(|durt|ddkrd|S|S)NrGr@z {} )r}rHr])stringr8r8r9maybe_start_on_next_line!s z,pretty_str..maybe_start_on_next_liner)rrrrc3s|] }|duVqdSrMr8)rOfieldZ mutable_dictr8r9rS4s  zpretty_str..zNo CUDArrrrz[{}] z[conda] r)r)rr)r)rr8) _asdictrallrorrrrsrr]rr env_info_fmt) ZenvinforrrrrZdynamic_cuda_fieldsZall_cuda_fieldsZall_dynamic_cuda_fields_missingrr8rr9 pretty_strs@          rcCs ttSrM)rrr8r8r8r9get_pretty_env_infoRs rcstdt}t|tr\ttdr^ttjdr`tjjjtj dkrbt j rdfddt D}t|t j jd}t j |}tj|d}d ||d }t|tjd dSdSdSdSdSdS) Nz%Collecting environment information...utils_crash_handlerrcsg|] }tj|qSr8)r r|r^)rOdumpZ minidump_dirr8r9r^szmain..)rz%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)printrrorprrrZDEFAULT_MINIDUMP_DIRrrr r|rlistdirmaxgetctimedatetime fromtimestampstrftimer]r")r6dumpsZlatestctimeZ creation_timemsgr8rr9mainVs     r__main__rM)2rr-rAr'rr collectionsrrro ImportError NameErrorAttributeErrorOSErrorrrZrr:r?rFrIr`rdrfrgrlrvrwrrirr,rrrrrrrrrrrrr0rrrr__name__r8r8r8r9sh     %[   9!K