U Mh=[@sddlZddlZddlZddlZddlZddlZddlmZzddlZdZ Wn e e e e fk rldZ YnXeddddd d d d d dddddddddddddddddgZdd d!d"d#d$d%d&hZdd d'd(d%d&d)hZd*d+Zd,d-Zd.d/Zd0d1Zdgd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Zd@dAZdBdCZdDdEZdFdGZdHdIZ dJdKZ!dLdMZ"dNdOZ#dPdQZ$dRdSZ%dTdUZ&dhdVdWZ'dXdYZ(dZd[Z)d\dZ*d]d^Z+d_,Z-d`daZ.dbdcZ/dddeZ0e1dfkre0dS)iN) 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_infotorchnumpyZ cudatoolkitZsoumithZmklZmagmaZtritonZoptreeZmypyZflake8Zonnxc Cszt|tkrdnd}tj|tjtj|d}|\}}|j}tdkrLd}nt }| |}| |}|| | fS)z%Return (return-code, stdout, stderr).TF)stdoutstderrshellwin32oem) typestr subprocessPopenPIPE communicate returncode get_platformlocalegetpreferredencodingdecodestrip) commandr!p raw_outputZraw_errrcencoutputerrr7I/var/www/html/venv/lib/python3.8/site-packages/torch/utils/collect_env.pyrunHs     r9cCs||\}}}|dkrdS|S)zIRun command using run_lambda; reads and returns entire output if rc is 0.rNr7 run_lambdar0r3out_r7r7r8run_and_read_allXsr>cCs<||\}}}|dkrdSt||}|dkr2dS|dS)zIRun command using run_lambda, returns the first regex match if it exists.rN)researchgroup)r;r0regexr3r<r=matchr7r7r8run_and_parse_first_match`s rEcCs(||\}}}|dkrdS|ddS)zKRun command using run_lambda and returns first line if output is not empty.rN )splitr:r7r7r8run_and_return_first_linejsrHcsRdkr ttjdd}t|d|}|dkr6|Sdfdd|DS)NZ CONDA_EXEcondaz{} listrFc3s2|]*dstfddDrVqdS)#c3s|]}|kVqdSNr7.0nameliner7r8 ~sz/get_conda_packages...N) startswithanyrMpatternsrOr8rQzs z%get_conda_packages..)DEFAULT_CONDA_PATTERNSr environgetr>formatjoin splitlines)r;rVrIr<r7rUr8get_conda_packagesrsr]cCs t|ddS)Nz gcc --versionzgcc (.*)rEr;r7r7r8get_gcc_versionsr`cCs t|ddS)Nzclang --versionzclang version (.*)r^r_r7r7r8get_clang_versionsracCs t|ddS)Nzcmake --versionz cmake (.*)r^r_r7r7r8get_cmake_versionsrbcCs,tdkrd}t||dSt}t||dS)Ndarwinzkextstat | grep -i cudazcom[.]nvidia[.]CUDA [(](.*?)[)]zDriver Version: (.*?) )r+rEget_nvidia_smi)r;cmdsmir7r7r8get_nvidia_driver_versions rgcCstdks&trttjdrtjjdk rtr~tjr~tjjdk rjtjd}t|drdd |j }qnd}nd}tj d|SdSt }t d}||d \}}}|dkrdSt |d|S) Nrchipr gcnArchNamez ({})ZNoGCNArchNameOnOldPyTorchz \(UUID: .+?\)z -L)r+TORCH_AVAILABLEhasattrrversionrhcuda is_availableZget_device_propertiesrZriZget_device_namerdr@compilesub)r;propZgcnArchrfZ uuid_regexr3r<r=r7r7r8 get_gpu_infos &    rscCs t|ddS)Nznvcc --versionzrelease .+ V(.*)r^r_r7r7r8get_running_cuda_versionsrtc Cs$tdkrDtjdd}tjdd}tj|dd}d||}ntd krTd }nd }||\}}}t|d ks|d kr|d krtjd}|dk rtj|rtj |SdSt } | dD]&} tj | } tj| r| | q| 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:\WindowsZ CUDA_PATHz %CUDA_PATH%System32wherez{} /R "{}\bin" cudnn*.dllrcz ls /usr/local/cuda/lib/libcudnn*z7ldconfig -p | grep libcudnn | rev | cut -d" " -f1 | revrr?Z CUDNN_LIBRARYNrFz!Probably one of the following: {})r+r rXrYpathr[rZlenisfilerealpathsetrGaddsorted) r; system_rootZ cuda_pathZ where_cmdZ cudnn_cmdr3r<r=lZ files_setfnfilesresultr7r7r8get_cudnn_versions4        rcCs|d}tdkrxtjdd}tjdd}tj|dd|}tj|d |}||g}|D]}tj|rXd |}qxqX|S) Nz nvidia-smir"rurvZ PROGRAMFILESzC:\Program FileszNVIDIA CorporationZNVSMIrwz"{}")r+r rXrYryr[existsrZ)rfrZprogram_files_rootZ legacy_pathnew_pathZsmisZ candidate_smir7r7r8rds   rdcCspd\}}}tdkr$|d\}}}n2tdkr>|d\}}}ntdkrV|d\}}}d}|d krh|}n|}|S) N)rrjrjlinuxZlscpur"zwmic cpu get Name,Manufacturer,Family,Architecture,ProcessorType,DeviceID, CurrentClockSpeed,MaxClockSpeed,L2CacheSize,L2CacheSpeed,Revision /VALUErcz"sysctl -n machdep.cpu.brand_stringNoner)r+)r;r3r<r6rr7r7r8 get_cpu_info/s    rcCsJtjdrdStjdr dStjdr0dStjdr@dStjSdS)Nrr"cygwinrc)sysplatformrRr7r7r7r8r+@s    r+cCs t|ddS)Nzsw_vers -productVersionz(.*)r^r_r7r7r8get_mac_versionMsrcCsBtjdd}tj|ddd}tj|dd}t|d||S)NrurvrwZWbemZwmicZfindstrz!{} os get Caption | {} /v Caption)r rXrYryr[r>rZ)r;rZwmic_cmdZ findstr_cmdr7r7r8get_windows_versionQsrcCs t|ddS)Nzlsb_release -azDescription:\t(.*)r^r_r7r7r8get_lsb_versionXsrcCs t|ddS)Nzcat /etc/*-releasezPRETTY_NAME="(.*)"r^r_r7r7r8check_release_file\srcCsddlm}t}|dks"|dkr*t|S|dkrTt|}|dkrFdSd||S|dkrt|}|dk rzd||St|}|dk rd||Sd||S|S) Nr)machiner"rrcz macOS {} ({})rz{} ({}))rrr+rrrZrr)r;rrrmdescr7r7r8get_osas$ rcCsddl}|SNrrrr7r7r8get_python_platformsrcCs$ddl}tdkrdSd|S)NrrN/A-)rr+r[libc_verrr7r7r8get_libc_versions rcsFdkr tfdd}tjddkr,dnd}|tjdg}||fS) zZReturn `pip list` output. Note: will also find conda-installed pytorch and numpy packages.Ncs.t|ddg}dfdd|DS)Nlistz--format=freezerFc3s(|] tfddDrVqdS)c3s|]}|kVqdSrKr7rLrOr7r8rQszCget_pip_packages..run_with_pip...N)rSrTrUrOr8rQsz9get_pip_packages..run_with_pip..)r>r[r\)pipr<rVr;r7r8 run_with_pipsz&get_pip_packages..run_with_pipr3Zpip3rz-mpip)DEFAULT_PIP_PATTERNSrrm executable)r;rVrrr<r7rr8get_pip_packagess rcCstjdd}|S)NZPYTORCH_CUDA_ALLOC_CONFrj)r rXrY)Z ca_configr7r7r8get_cachingallocator_configsrcCs2tr*tjr*tjtjdd}|SdSdS)NZCUDA_MODULE_LOADINGrjr)rkrrnroinitr rXrY)configr7r7r8get_cuda_module_loading_configs  rcCs"trddl}t|jjjSdSdS)Nrr)rkZtorch.backends.xnnpackr%backendsZxnnpackZenabled)rr7r7r8rscCsHt}t|\}}trtj}ttjj}ttj }tjj}t tjdrTtjj dkrbd}}} qdd} tj d} | | d}| | d} d}tjj }nd}}}}d}}} tjdd } t|} t||d | tjd t||t|tt|t|t|||| ||| t|tt|t|t |t!t"t#|d 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. rhNrcs"fdd|D}|r|dSdS)Ncs$g|]}|kr|dddqS)Nr?)rsplit)rMsprefixr7r8 sz;get_env_info..get_version_or_na..rrr7)cfgrZ_lstr7rr8get_version_or_nasz'get_env_info..get_version_or_narFz HIP RuntimeZMIOpen z{} ({}-bit runtime)r?)rrr r rrrrrrrrrrrrrr r rrr rrr)$r9rrkr __version__r%rmdebugrnrorlrhZ_CZ _show_configrGrreplacer]rrZmaxsize bit_lengthrrtrrsrgrrrr`rarbrrr)r;rZpip_list_output version_strZdebug_mode_strZcuda_available_strZcuda_version_strrrrrr sys_versionrr7r7r8 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}d d d }d!d d }dd}|||jd<dddg}|dg}tfdd|D}trtjs|r|D] } d| <q|jdkrdd<|||dd<|dd<dr|dd|j d<dr|ddd<|j d<t jfS)"NCould not collectcSs(|D]}||dk rq|||<q|SrKkeys)dct replacementkeyr7r7r8 replace_noness    z!pretty_str..replace_nonesYesNocSs<|D].}||dkr"|||<q||dkr|||<q|S)NTFr)rtruefalserr7r7r8 replace_bools&s      z!pretty_str..replace_bools [prepend]cs&|d}fdd|D}d|S)NrFcsg|] }|qSr7r7)rMrPtagr7r8r0sz/pretty_str..prepend..)rGr[)textrlinesZ updated_linesr7rr8prepend.s zpretty_str..prependNo relevant packagescSs|dk rt|dkr|S|Sr)rz)rrr7r7r8replace_if_empty3sz$pretty_str..replace_if_emptycSs(|dk r$t|ddkr$d|S|S)NrFr?z {} )rzrGrZ)stringr7r7r8maybe_start_on_next_line8s z,pretty_str..maybe_start_on_next_linerrrrc3s|]}|dkVqdSrKr7)rMfieldZ mutable_dictr7r8rQKszpretty_str..zNo CUDArrrrz[{}] z[conda] r)r)rr)r)r) _asdictrallrkrrnrorrZrr env_info_fmt) ZenvinforrrrrZdynamic_cuda_fieldsZall_cuda_fieldsZall_dynamic_cuda_fields_missingrr7rr8 pretty_strsF            rcCs 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. )rrr7r7r7r8get_pretty_env_infois rcstdt}t|trttdrttjdrtjjjtj dkrt j rfddt D}t|t j jd}t j |}tj|d}d ||d }t|tjd dS) Nz%Collecting environment information...utils_crash_handlerrcsg|]}tj|qSr7)r ryr[)rMdumpZ minidump_dirr7r8rszmain..)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)printrrkrlrrrZDEFAULT_MINIDUMP_DIRrrr ryrlistdirmaxgetctimedatetime fromtimestampstrftimerZr )r5dumpsZlatestctimeZ creation_timemsgr7rr8mainxs   r__main__)N)N)2rr,r@r&rr collectionsrrrk ImportError NameErrorAttributeErrorOSErrorrrWrr9r>rErHr]r`rarbrgrsrtrrdrr+rrrrrrrrrrrrr/rrrr__name__r7r7r7r8s       %[   H!K