GCP: невозможно подключиться к облачному TPU

Question

Я запустил экземпляр с TPU, следуя этому краткому руководству с помощью команды ctpu up, и смог успешно запустить учебник MNIST. Я вышел из облачной оболочки и вошел в свой vm, подключенный к TPU, используя консоль SSH, как объяснено здесь , когда я снова запускаю учебник MNIST, я получаю

RuntimeError: Cannot find any TPU cores in the system. Please double check Tensorflow master address and TPU worker(s).

Когда я запускаю ctpu ls, я получаю

#       Flock Name      Status
0:      my-tpu(*)  running

ctpu status команда дает

Your cluster is running!
Compute Engine VM:  RUNNING
Cloud TPU:          RUNNING

Я что-то упустил здесь?

Answer 1

ctpu передает это имя в виртуальную машину Compute Engine как переменную среды (TPU_NAME), но gcloud нет.

Укажите ваш TPU явно: используйте --tpu=my-tpu вместо --tpu=$TPU_NAME