标签:python device nvidia ssi 日志 igp config 使用 计算
查看是否用GPU跑的TensorFlow程序
第一种方法,直接输出日志法(推荐)
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
第二种方法,跑计算量打的程序,然后用nvidia-smi
查看GPU的使用情况
标签:python device nvidia ssi 日志 igp config 使用 计算
原文地址:https://www.cnblogs.com/youpeng/p/10887364.html