Очень часто люди, работающие над Google Colab, пытаются установить torch-xla с помощью следующей команды:

!pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.9-cp37-cp37m-linux_x86_64.whl

ИЗМЕНИТЬ:

По состоянию на 14 сентября 2021 года колесо torch-xla-1.9 было исправлено, как сообщается на вкладке "Проблема" GitHub ("#3121").

Итак, теперь любой может просто использовать

!pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.9-cp37-cp37m-linux_x86_64.whl

это без ошибок!!

Но если ошибки, подобные упомянутой ниже, повторяются, вы знаете, что делать!

И при попытке импортировать torch_xla получаем следующую ошибку:

--------------------------------------------------------------------
ImportError                       Traceback (most recent call last)
<ipython-input-3-5bff978a6cc6> in <module>()
      2 import torch.nn as nn
      3 import torchvision
----> 4 import torch_xla
      5 import torch_xla.core.xla_model as xm
      6 import torch_xla.debug.metrics as met
/usr/local/lib/python3.7/dist-packages/torch_xla/__init__.py in <module>()
     99 from ._patched_functions import _apply_patches
    100 from .version import __version__
--> 101 import _XLAC
    102 
    103
ImportError: /usr/local/lib/python3.7/dist-packages/_XLAC.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBase8data_ptrIN3c107complexIfEEEEPT_v

Как нам быть с ошибкой?

На данный момент кажется, что мы можем избавиться от этой ошибки, просто понизив версию PyTorch до torch-1.8.2+cpu.

Для соответствия версий нам нужно установить torch-xla-1.8

!pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.8-cp37-cp37m-linux_x86_64.whl

После этого импорт torch_xla дает

WARNING:root:Waiting for TPU to be start up with version pytorch-1.8
WARNING:root:Waiting for TPU to be start up with version pytorch-1.8
WARNING:root:TPU has started up successfully with version pytorch-1.8

Ссылка:

  1. https://github.com/pytorch/xla/issues/3121