作为我研发的一部分,我获得了对多个 TPU 的访问权限,但我找不到如何将它们分配在一起以用于我的培训目的的文档,无论是在节点方面还是在代码方面。文档说ctpu up -zone MY_ZONE_CHOICE
,但是这个命令只分配一个 TPU。而且,如果我想使用多个 TPU,我应该在代码中添加哪些类似的更改?到目前为止,我已经使用这个调用tf.contrib.cluster_resolver.TPUClusterResolver()
来检查 TPU,应该改变什么(如果有的话)来检查我是否可以访问多个 TPU?
问问题
58 次