google-cloud-platform - 运行 pytorch/xla 时缺少 XLA 配置
问题描述
我正在尝试使用 Pytorch/XLA 运行 GCP TPU,我正在使用带有 debian-9-torch-xla-v20200818 映像的 VM,我启动 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
我尝试了一切,但似乎没有任何效果。
解决方案
推荐阅读
- c# - 使用 EF Core 继承时如何避免重复属性投影?
- powerpivot - 具有来自多个表的组合度量的 Power Pivot
- .net - ILMerge / ILRepack dll 无法使用 Dynamics 365 插件注册工具注册
- node.js - Sinon mock 正在调用真正的 AWS Secrets Manager
- python - 为什么一个类可以调用另一个类的魔法方法?
- google-my-business - 如何解决“Google 我的商家”位置重复问题?
- sql - 根据第一次查询的结果将列添加到 SQL 结果
- python - Unicode 转义 Python
- c# - Windows Application Packaging Project
- javascript - Discord.js command handler bot doesn't respond