o ht@s,ddlZddlZddlZddefddZdS)NFupgradecCsd}dtjvrdttjdv}|rD|r!gd}tj|ddtjd}|d| d }d |d }d d |g}tj|dddSt d)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_cmdrP/var/www/html/ai/venv/lib/python3.10/site-packages/accelerate/utils/torch_xla.py install_xlas    r)F)importlib.metadatarr r boolrrrrrs