Очень часто люди, работающие над 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