vhvddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl ZddZd d dZy) ) annotationsN)version)lib) xla_bridgecj tjdgjS#t$rYywxYw)Nz nvidia-smi) subprocess check_outputdecode ExceptionT/opt/face_recognition/venv/lib/python3.12/site-packages/jax/_src/environment_info.pytry_nvidia_smirs5  " "L> 2 9 9 ;;  s #& 22ctjjdd}tjdtj dt jdtj d|dtjdjd tjd tjd tjd tj }t"j$j'D]"\}}|j)d s|d|d|z }$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.   z jax: z jaxlib: z numpy: z python: z device info: r-z, z! local devices" process_count: z platform: JAX_=z $ nvidia-smi )sysrreplacetextwrapdedent __version__r version_strnpxbdevices device_kind device_countlocal_device_count process_countplatformunameosenvironitems startswithrprint) return_stringpython_versioninfokeyvalue nvidia_smis rprint_environment_infor0#s;;;&&tS1.      ??  >>    Q++,Aboo.?-@2CXCXCZB[\""$%& ^^  " #$JJ$$&!jc5 ~~f 3%q  d!* : --D K ;r )return str | None)F)r*boolr1r2) __future__rr%r#rrrjaxrjax._srcrrrnumpyrrr0r r rr8s/#  %r