查看是否用GPU跑的TensorFlow程序
第一种方法，直接输出日志法（推荐）
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))第二种方法，跑计算量打的程序，然后用nvidia-smi查看GPU的使用情况
查看是否用GPU跑的TensorFlow程序
猜你喜欢
转载自www.cnblogs.com/youpeng/p/10887364.html
今日推荐
周排行