o
    
j+                     @   s   d Z ddlZddlZddlZddlZddlZe jejdejdg7  _ddlZddl	Z	dd Z
dd Zd	d
 Zdd ZedkrHe  dS dS )z0Get detailed info about the working environment.    Nz...c                   C   s$   t  t  t  t  t  dS )N)OSarchitectureversion	processorpython)platformsystemr   r   r   python_version r   r   K/home/kuhnn/.local/lib/python3.10/site-packages/TTS/bin/collect_env_info.pysystem_info   s   r   c                   C   s*   dd t tj D tj tjjdS )Nc                 S   s   g | ]}t j|qS r   )torchcudaget_device_name).0ir   r   r   
<listcomp>   s    zcuda_info.<locals>.<listcomp>)GPU	availabler   )ranger   r   device_countis_availabler   r   r   r   r   	cuda_info   s   r   c                   C   s   t jtjtjjtjdS )N)numpyPyTorch_versionPyTorch_debugTTS)r   __version__r   r   debugr   r   r   r   r   package_info!   s
   r    c                  C   s*   t  t t d} ttj| ddd d S )N)SystemCUDAPackages   T)indent	sort_keys)r   r   r    printjsondumps)detailsr   r   r   main*   s   r+   __main__)__doc__osr   sysr   r   pathabspathr(   r   r   r   r    r+   __name__r   r   r   r   <module>   s     "
	
