Как выделить несколько TPU и настроить код для запуска на всех них - PullRequest
0 голосов
/ 04 марта 2019

как часть моих исследований и разработок мне был предоставлен доступ к нескольким TPU, но я не могу найти документацию о том, как распределить их вместе для моих учебных целей, как для узлов, так и для кода.В документации сказано ctpu up -zone MY_ZONE_CHOICE, но эта команда выделяет только один TPU.И, похоже, какие изменения я должен добавить в свой код, если я хочу использовать несколько TPU?До сих пор я использовал этот вызов tf.contrib.cluster_resolver.TPUClusterResolver() для проверки TPU, что нужно изменить (если таковые имеются), чтобы проверить, могу ли я получить доступ к нескольким TPU?

...