gt,ddlZddlZddlZddefdZdS)NFupgradecd}dtjvr3dttjdv}|r{|rgd}t j|dt jd}|d| d }d |d }d d |g}t j|ddStd)a Helper function to install appropriate xla wheels based on the `torch` version in Google Colaboratory. Args: upgrade (`bool`, *optional*, defaults to `False`): Whether to upgrade `torch` and install the latest `torch_xla` wheels. Example: ```python >>> from accelerate.utils import install_xla >>> install_xla(upgrade=True) ``` FIPythonz google.colab)pipinstallz-UtorchT)checkrN.zBhttps://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-z-cp37-cp37m-linux_x86_64.whlrrz1`install_xla` utility works only on google colab.) sysmodulesstr get_ipython subprocessrun importlibmetadataversionrindex RuntimeError)rin_colabtorch_install_cmd torch_versiontorch_version_trunc xla_wheelxla_install_cmds V/var/www/html/ai-engine/env/lib/python3.11/site-packages/accelerate/utils/torch_xla.py install_xlars HCK!SY)?)K)K)M)M%N%NN P  : A A A  N,D 9 9 9 9!*227;; +,Gm.B.B3.G.G,GHKYlKKK  )Y7d333333NOOO)F)importlib.metadatarrr boolrrrr"sZ PPPPPPPPr