使用 TPU 的深度学习虚拟机

【问题标题】：Deep learning vm with TPU使用 TPU 的深度学习虚拟机
【发布时间】：2021-02-25 16:36:57
【问题描述】：

请帮助我使用 TPU 设置 深度学习 vm (dlvm)，我已经在同一个区域中创建了 dlvm 和 TPU 节点，现在当我访问 jupyter notebook，它没有使用 TPU 节点，而只使用了 vm 配置（CPU 和 Ram）。我正在寻找将 vm 与 TPU 链接的选项，但找不到。

【问题讨论】：

标签： google-cloud-platform

【解决方案1】：

您需要使用--tpu-only flag:

仅分配一个 Cloud TPU；仅当您已有可用的 VM 时才使用此选项。

之后，当您运行训练时，指定您的 TPU，它将被使用。

Example:

python3 mnist_main.py \
  --tpu=$TPU_NAME \
  --model_dir=$MODEL_DIR \
  --data_dir=$DATA_DIR \
  --train_epochs=10 \
  --distribution_strategy=tpu \
  --download

【讨论】：