我想看看哪些操作在哪个设备上运行。在 TensorFlow 1.n 中，可以使用会话配置 (log_device_placement)。TF 2.n 中是否有类似的操作？
问问题
822 次
1 回答
2
为了查找哪些操作在哪个设备上运行，请使用
tf.debugging.set_log_device_placement(True)
作为导入 TensorFlow 后的第一行。您可以在此处找到更多详细信息
于 2020-05-03T16:08:59.767 回答