如何解决即使在 PyTorch Lightning 中连接了 TPU,也会出现“错误配置错误:未找到 TPU 设备”
过去几个小时一直因为一个问题而感到沮丧,虽然这很可能是我自己开始的一个问题哈哈。
我正在尝试连接到 Colab 中的 TPU。我很确定我已经把所有进口的东西都弄下来了。我的代码是 here. 我还没有完全设置好所有内容,所以整个文档都不起作用,但您应该能够看到我尝试连接 TPU。
我在 1.5.0 版中运行 Pytorch,在 0.6.0 版中运行 Torchvision,因为我发现我无法在 1.5.0 之后的任何版本中安装 XLA。我在 20200325 版本中运行 XLA。
这张图片看起来很混乱:它表明我们与 xla: 1 有联系,但是当我尝试在训练器中标记它时,我收到一条错误消息,说找不到 TPU。
如果有人能帮助我,那就太棒了。
谢谢, 一个
解决方法
我遇到了同样的问题,这些步骤解决了问题。
- 遵循 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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。