问题描述
过去几个小时一直因为一个问题而感到沮丧,虽然这很可能是我自己开始的一个问题哈哈。
我正在尝试连接到 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