jupyter notebook tensorflow打印device信息
juypter notebook中直接使用log_device_placement=True打印不出来device信息
# Creates a graph.
with tf.device('/device:CPU:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = t...
原创
2018-08-11 14:05:23 ·
945 阅读 ·
0 评论