s7ev3n
6/9/2018 - 8:23 AM

查看运算是在cpu还是gpu

查看运算是在cpu还是gpu

sess = tf.Sesstion(config=tf.ConfigProto(log_device_placement=True))