“错误配置错误：未找到 TPU 设备”即使在 PyTorch Lightning 中连接了 TPU答案

【问题标题】："MisconfigurationError: No TPU devices were found" even when TPU is connected in PyTorch Lightning“错误配置错误：未找到 TPU 设备”即使在 PyTorch Lightning 中连接了 TPU
【发布时间】：2021-03-30 22:24:00
【问题描述】：

在过去的几个小时里一直因为一个问题而感到沮丧，尽管这很可能是我自己开始的问题哈。

我正在尝试连接到 Colab 中的 TPU。我很确定我已经把所有进口的东西都搞定了。我的代码是 here. 我还没有完全设置好所有内容，所以整个文档都无法正常工作，但您应该能够看到我尝试 TPU 连接。

我在 1.5.0 版本中运行 Pytorch，在 0.6.0 版本中运行 Torchvision，因为我发现我无法使用 1.5.0 之后的任何东西安装 XLA。我在 20200325 版本中运行 XLA。

这张图片看起来很混乱：它表明我们与 xla: 1 有联系，但是当我尝试在训练器中标记它时，我收到一条错误消息，提示找不到 TPU。

如果有人可以帮助我，那就太棒了。

谢谢，一个

【问题讨论】：

只是想知道 - 您是否尝试过使用文档 here 安装 Torch/Torch-XLA/PyTorch-Lightning？如您所述，Torch XLA 似乎已正确安装，但看起来 PyTorch-Lightning 需要一些额外的步骤。

标签： pytorch tpu pytorch-lightning

【解决方案1】：

我遇到了同样的问题，这些步骤解决了问题。

关注 PyTorch-Lightning 文档：TPU SUPPORT
添加另一个笔记本单元格：

%%capture
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py > /dev/null
!python pytorch-xla-env-setup.py --version nightly --apt-packages libomp5 libopenblas-dev > /dev/null
!pip install pytorch-lightning > /dev/null

【讨论】：