我想使用下面的代码来研究设备在拉伸板中的位置,以生成摘要中的图形
# Build the summary operation based on the TF collection of Summaries但是当选择‘设备放置’在拉伸板,所有的节点被分配给‘未知的设备’。我需要以其他方式转储设备的位置吗?
我在一台安装并启用了图形处理器的机器上使用了,它似乎只在预测阶段使用了CPU。我在.predict()部分看到的设备放置日志如下:
2020-09-01 06:08:19.085400: I tensorflow/core/common_runtime/eager/execute.ccop RangeDataset in device /job:localhost/replica:0/task:0/devic
我试图使用Estimator-based Tensorflow模型实现TPUEstimator API,但没有成功。它在训练中遇到了一个错误:
InvalidArgumentError (see above for traceback): No OpKernel was registered to support OpCrossReplicaSum_5 = CrossReplicaSum[T=DT_FLOAT](gradients/dense_2/BiasAdd_grad/tuple/control_dependenc