查看是否用GPU跑的TensorFlow程序

查看是否用GPU跑的TensorFlow程序

第一种方法，直接输出日志法（推荐）

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

第二种方法，跑计算量打的程序，然后用nvidia-smi查看GPU的使用情况

posted @ 2019-05-18 22:13  Youpeng  阅读(4984)  评论(0)  编辑  收藏  举报