
    9i
                     8    S SK JrJr  S rS rS rS rS rS rg)	   )is_dep_availablerequire_depsc                  R    SS K n U R                  5       nUR                  S5      S   $ )N    :)paddle
get_devicesplit)r   
device_strs     Q/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/utils/env.pyget_device_typer      s)    ""$JC ##    c                      SS K n U R                  nSU;   a  UR                  S5      u  pOS nUR                  S5      n[        U5      S:X  d   e[	        [
        U5      u  p4nU(       a  X4XR4$ X4US 4$ )Nr   -.   )r   __version__r
   lenmapint)r   versiontagmajor_vminor_vpatch_vs         r   get_paddle_versionr      sz      G
g~}}S)mmC Gw<1 #C 1Gg
--$..r   c                      SS K n U R                  R                  5       nUS:X  a  g [        [	        [
        UR                  S5      5      5      $ Nr   Falser   )paddle.versionr   cudatupler   r   r
   )r   cuda_versions     r   get_paddle_cuda_versionr$   *   s?    >>&&(LwS,,,S1233r   c                      SS K n U R                  R                  5       nUS:X  a  g [        [	        [
        UR                  S5      5      5      $ r   )r    r   cudnnr"   r   r   r
   )r   cudnn_versions     r   get_paddle_cudnn_versionr(   3   s?    NN((*MS---c2344r   c                  *   [        S5      (       a:  SS Kn U R                  R                  5       =(       a    U R	                  5       (       + $ [        S5        SS KnSS KnUR                  R                  5       =(       a    UR                  R                  $ Npaddlepaddler   torch)r   paddle.devicedeviceis_compiled_with_cudais_compiled_with_rocmr   
torch.cudatorch.versionr!   is_availabler   )r   r,   s     r   is_cuda_availabler4   ?   sl    '' MM//1X&:V:V:X6X	

 	W zz&&(?U]]-?-??r   c                      S n [        5       (       ac  [        S5      (       a*  SS KnUR                  R                  R                  5       n U $ [        S5        SS KnUR                  R                  5       n U $ r*   )r4   r   r-   r.   r!   get_device_capabilityr   r1   )capr   r,   s      r   get_gpu_compute_capabilityr8   R   sa    
CN++ --$$::<C J !**224CJr   N)	depsr   r   r   r   r$   r(   r4   r8    r   r   <module>r;      s)    1$/"45@&r   