摘要: 首先检测TPU存在: tpu = tf.distribute.cluster_resolver.TPUClusterResolver() #如果先前设置好了TPU_NAME环境变量,不需要再给参数. tpu的返回值为1 or 0 ,1则检测到了TPU. tf.config.experimental_ 阅读全文
posted @ 2020-02-29 13:26 dynmi 阅读(1965) 评论(0) 推荐(0) 编辑