Ошибка инициализации TPU (отсутствует OpKernel), несмотря на активную среду выполнения TPU в Kaggle ⇐ Python

[Anonymous]

Я постоянно сталкиваюсь с проблемой при попытке инициализировать TPU в своем ноутбуке. Я уже подтвердил, что:

• Моя учетная запись проверена.
• Ускоритель ноутбука настроен на TPU.
• Моя квота TPU в настоящее время доступна.

введите описание изображения здесь
введите описание изображения здесь
введите описание изображения здесь
Однако стандартный код инициализации постоянно выдает ошибку NotFoundError, поскольку необходимый OpKernel отсутствует. Я подозреваю, что это проблема с конфигурацией среды самой платформы.
Кто-нибудь недавно сталкивался с этой конкретной ошибкой OpKernel не зарегистрирован при использовании среды выполнения TPU и нашел обходной путь?
Сведения о коде и ошибке
Использованный код:

Код: Выделить всё

import tensorflow as tf
# Detect and initialize TPU
tpu = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='local')
tf.tpu.experimental.initialize_tpu_system(tpu)
# Create TPU distribution strategy
strategy = tf.distribute.TPUStrategy(tpu)
print("TPU initialized successfully.")

Фрагмент трассировки:

Код: Выделить всё

InvalidArgumentError: No OpKernel was registered to support Op 'ConfigureDistributedTPU' used by {{node ConfigureDistributedTPU}}
...
Registered devices: [CPU]
Registered kernels:


During handling of the above exception, another exception occurred:
NotFoundError: TPUs not found in the cluster. Failed in initialization: No OpKernel was registered to support Op 'ConfigureDistributedTPU'...

Основное наблюдение:
Вывод показывает Зарегистрированные устройства: [ЦП], что подтверждает, что среда не обнаруживает активный ускоритель TPU на уровне программного обеспечения TensorFlow.
Будем очень признательны за любую помощь или известные обходные пути\! Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/798 ... -in-kaggle