常见问题解答 - Cloud TPU

本文档包含有关 Cloud TPU 的常见问题解答列表。它 分为以下几个部分：

  1. 框架无关的常见问题解答 - 无论您使用的是哪种机器学习框架，都可以查看与使用 Cloud TPU 相关的问题。
  2. JAX 常见问题解答 - 有关将 Cloud TPU 与 JAX 搭配使用的问题。
  3. PyTorch 常见问题解答 - 有关将 Cloud TPU 与 PyTorch 搭配使用的问题。

与框架无关的常见问题解答

如何检查 Cloud TPU 虚拟机上哪个进程正在使用 TPU？

在 Cloud TPU 虚拟机上运行 sudo lsof -w /dev/accel*，以输出进程 ID 和 有关使用 TPU 进程的其他信息。

如何向 Cloud TPU 虚拟机添加永久性磁盘卷？

如需了解详情，请参阅将永久性磁盘添加到 TPU 虚拟机

使用 TPU VM 进行训练时支持/推荐使用哪些存储方案？

如需了解详情，请参阅 Cloud TPU 存储方案。

JAX 常见问题解答

如何判断我的程序是否在使用 TPU？

您可以通过以下几种方式来仔细检查 JAX 是否使用 TPU：

  1. 使用 jax.devices() 函数。例如：

    assert jax.devices()[0].platform == 'tpu'

  2. 分析程序并验证配置文件是否包含 TPU 操作。有关 相关信息，请参阅剖析 JAX 程序的性能

如需了解详情，请参阅 JAX 常见问题解答

Pytorch 常见问题解答

如何判断我的程序是否在使用 TPU？

您可以运行以下 Python 命令：

>>> import torch_xla.core.xla_model as xm
>>> xm.get_xla_supported_devices(devkind="TPU")

验证是否可以看到任何 TPU 设备。