如何在 TensorFlow 2.n 中启用设备放置日志记录 (log_device_placement)
How do I enable device placement logging (log_device_placement) in TensorFlow 2.n
我想看看在哪个设备上进行了哪些操作运行。在 TensorFlow 1.n 中，可以使用会话配置 (log_device_placement)。 TF里有没有类似的操作2.n?
为了在哪个设备上找到哪些操作运行，使用
tf.debugging.set_log_device_placement(True)
作为导入 TensorFlow 后的第一行。您可以找到更多详细信息 here
我想看看在哪个设备上进行了哪些操作运行。在 TensorFlow 1.n 中，可以使用会话配置 (log_device_placement)。 TF里有没有类似的操作2.n?
为了在哪个设备上找到哪些操作运行，使用
tf.debugging.set_log_device_placement(True)
作为导入 TensorFlow 后的第一行。您可以找到更多详细信息 here