VpfpddlmZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddl Z d dZ ddd ZdS)) annotationsN)version)lib) xla_bridgereturn str | Nonect tjdgS#t$rYdSwxYw)Nz nvidia-smi) subprocess check_outputdecode ExceptionY/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/environment_info.pytry_nvidia_smirsH  "L> 2 2 9 9 ; ;;  44s &) 77F return_stringboolctjdd}tjdd5t tjtjdd}dddn #1swxYwYtj dtj d tj d tj d |d tjd tjd|dtjdt#jd}t'}|r|d|zz }|r|St)|S)zReturns a string containing local environment & JAX installation information. This is useful information to include when asking a question or filing a bug. Args: return_string (bool) : if True, return the string rather than printing to stdout.   ) threshold edgeitemsNz jax: z jaxlib: z numpy: z python: z jax.devices (z total, z local): z process_count: z platform: z $ nvidia-smi )sysrreplacenp printoptionsstrarrayrdevicestextwrapdedent __version__r version_str device_countlocal_device_count process_countplatformunamerprint)rpython_version devices_shortinfo nvidia_smis rprint_environment_infor1"s;&&tS11. a000JJ!3!5!56677??bIIMJJJJJJJJJJJJJJJ    ? >   '))  4>3P3R3R  ^k  *,, ^    $*. : --D K ;;sAB  BB)rr)F)rrrr) __future__rr*r rr#jaxrjax._srcrrnumpyrrr1rrrr6s#""""" r