为什么PyTorch中的Google上的TPU没有被检测到？

Question

我正在使用google和PyTorch。我把硬件加速器调到TPU了。

这一行代码显示没有检测到cuda设备：

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device)

Answer 1

TPU不是CUDA设备。请看一下这个Colab，它展示了如何在TPU上运行一些PyTorch代码。