查看是否用GPU跑的TensorFlow程序
第一种方法，直接输出日志法（推荐）
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
第二种方法，跑计算量打的程序，然后用nvidia-smi查看GPU的使用情况
查看是否用GPU跑的TensorFlow程序
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。