查看是否用GPU跑的TensorFlow程序
第一种方法,直接输出日志法(推荐)
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))imp查看是否用GPU跑的TensorFlow程序
第一种方法,直接输出日志法(推荐)
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))imp