U
|
©r_é ã @ sØ d Z ddlZddlZddlZddlZddlZddlmZ ddlZddl Z
|
ddl Z ddl Z ddl m Z dgZdd Zdd Zd
|
d Zd d Zed krÔz ddlZW n ek
|
r¼ ee Y nX ddlmZ ee dS )z:
|
@author: xingyu liao
|
@contact: sherlockliao01@gmail.com
|
é N)Ú defaultdict)ÚtabulateÚcollect_env_infoc C s@ zdd l } | j ¡ W S tk
|
r: ddlm} | Y S X d S )Nr )Úget_pretty_env_info)Útorch.__config__Ú
|
__config__ÚshowÚ ImportErrorZtorch.utils.collect_envr )Útorchr © r ú./utils/collect_env.pyÚcollect_torch_env s r c C s d} | t j | d¡fS )NÚFASTREID_ENV_MODULEz <not set>)ÚosÚenvironÚget)Úvar_namer r r Úget_env_module% s r c C s¸ zt j | dd¡}t j |¡rtjd ||¡dd}| d¡ ¡ d¡}g }|D ]$}t
|
d|¡d }| | d
|
¡¡ qPt t|}d |¡W S |d W S W n tk
|
r² | Y S X d S ) NÚbinÚ cuobjdumpz'{}' --list-elf '{}'T)Úshellzutf-8Ú
|
z \.sm_[0-9]*\.r Ú.z, z; cannot find cuobjdump)r ÚpathÚjoinÚisfileÚ
|
subprocessÚ check_outputÚformatÚdecodeÚstripÚsplitÚreÚfindallÚappendÚsortedÚsetÚ Exception)Ú CUDA_HOMEZso_filer ÚoutputÚsmÚliner r r Ú detect_compute_compatibility* s"
|
ÿ r, c C s2 t j ¡ } t j}ddlm} d}ttt| d¡d d
dkrjddlm
|
} t t j dd d k rj|d k rjd }| or| }g }| d
|
tjf¡ | d tj d d ¡f¡ | dtjf¡ z.dd l}| d|jd tj |j¡ f¡ W n tk
|
rü | d¡ Y nX | t ¡ | d|d tj t j¡ f¡ | dt j jf¡ | d| f¡ | rtt}t t j ¡ D ] } |t j | ¡ t | ¡ qf| ¡ D ]"\}
|
} | dd !| ¡ |
|
f¡ q|rÎ| dt |f¡ n4| dt |f¡ tj" #dd ¡} | r| d| f¡ | dt$jf¡ z|| dt t%jd tj t%j¡ f¡ |rz*t&j' (d ¡j)} t*|| }| d |f¡ W n tk
|
r | d ¡ Y nX W n t+k
|
r® | d ¡ Y nX z dd l,}| d |jf¡ W n tk
|
râ Y nX z dd l-}| d!|jf¡ W n tk
|
r Y nX t.|d }|t/ 7 }|S )"Nr )r( Fr é )é é )Ú ROCM_HOMEÚhipTz sys.platformÚPythonr Ú ÚnumpyÚfastreidz @)r5 zfailed to importZPyTorchzPyTorch debug buildz GPU availablezGPU ú,r0 r( ZTORCH_CUDA_ARCH_LISTZPillowÚ torchvisionútorchvision._Cztorchvision arch flags)r8 zfailed to find)r7 ÚunknownÚfvcoreÚcv2)0r
|
ÚcudaÚ is_availableÚ __version__Ztorch.utils.cpp_extensionr( ÚtupleÚmapÚintr! r0 ÚgetattrÚversionr$ ÚsysÚplatformÚreplaceÚnpr5 r r ÚdirnameÚ__file__r r Údebugr ÚlistÚrangeÚ device_countÚget_device_nameÚstrÚitemsr r r ÚPILr7 Ú importlibÚutilÚ find_specÚoriginr, ÚAttributeErrorr: r; r r )Zhas_gpuZ torch_versionr( Zhas_rocmr0 Úhas_cudaÚdatar5 ÚdevicesÚkÚnameZdevidsZcuda_arch_listZ torchvision_CÚmsgr: r; Zenv_strr r r r ? s
|
|
ÿ þÿ
|
|
Ú__main__)r )Ú__doc__rR r r" r rD Ú collectionsr rQ r4 rG r
|
r7 r Ú__all__r r r, r Ú__name__Ú
|
detectron2r ÚprintZfastreid.utils.collect_envr r r r Ú<module> s. W
|
