U 0-et@s,ddlZddlZddlZdedddZdS)NF)upgradecCsd}dtjkr$dttjdk}|r|rFddddg}tj|dd tjd}|d | d }d |d }dd|g}tj|dd nt dd S)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) ``` FZIPythonz google.colabpipinstallz-UZtorchT)checkN.zBhttps://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-z-cp37-cp37m-linux_x86_64.whlz1`install_xla` utility works only on google colab.) sysmodulesstr get_ipython subprocessrun importlibmetadataversionrindex RuntimeError)rZin_colabZtorch_install_cmdZ torch_versionZtorch_version_truncZ xla_wheelZxla_install_cmdr[/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/accelerate/utils/torch_xla.py install_xlas     r)F)importlib.metadatar r rboolrrrrrs