运行pytorch/xla时缺少XLA配置

Question

我尝试使用Pytorch/XLA运行GCP TPU，我使用具有debian-9-torch-xla-v20200818镜像的虚拟机，我启动TPU并使用ctpu状态检查它是否正在运行，该状态显示CPU和TPU都在运行，然后我激活torch-xla-nightly环境，但当我尝试调用以下简单代码时：

import torch
import torch_xla
import torch_xla.core.xla_model as xm

dev = xm.xla_device()
t1 = torch.ones(3, 3, device = dev)
print(t1)

出现以下错误：

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/anaconda3/envs/torch-xla-nightly/lib/python3.6/site-packages/torch_xla/core/xla_model.py", line 231, in xla_device
devkind=devkind if devkind is not None else None)
File "/anaconda3/envs/torch-xla-nightly/lib/python3.6/site-packages/torch_xla/core/xla_model.py", line 136, in get_xla_supported_devices
 xla_devices = _DEVICES.value
File "/anaconda3/envs/torch-xla-nightly/lib/python3.6/site-packages/torch_xla/utils/utils.py", line 32, in value
self._value = self._gen_fn()
File "/anaconda3/envs/torch-xla-nightly/lib/python3.6/site-packages/torch_xla/core/xla_model.py", line 18, in <lambda>
_DEVICES = xu.LazyProperty(lambda: torch_xla._XLAC._xla_get_devices())
RuntimeError: tensorflow/compiler/xla/xla_client/computation_client.cc:274 : Missing XLA configuration

我什么都试过了，但似乎都不管用。

Answer 1

看一看this link，因为它似乎与这个问题有关。也许您没有设置导出：(vm)$ XRT_TPU_CONFIG XRT_TPU_CONFIG="tpu_worker;0;$TPU_IP_ADDRESS:8470“按照here的说明操作，您应该没问题。

Answer 2

如果您看到正确设置了XRT_TPU_CONFIG，另一种可能是您忘记使用适当的scopes启动实例

gcloud compute instances create ... --scopes=https://www.googleapis.com/auth/cloud-platform