查看运行tensorflow程序是否使用的GPU

查看运行tensorflow程序是否使用的GPU

查看是否用GPU跑的TensorFlow程序

第一种方法,直接输出日志法(推荐)

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

打印出这个就代表使用了cpu:
查看运行tensorflow程序是否使用的GPU_第1张图片

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

你可能感兴趣的:(tensorflow,python)