如何查看tensorflow是GPU还是CPU版本

AIcomes手机认证

运行下面python代码，查看输出结果

1. sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
   
2. sess.close()

复制代码

如果输出结果里面有显卡型号信息（如下图），则安装的tensorflow为GPU版，否则为CPU版




参考：https://stackoverflow.com/questions/38009682/how-to-tell-if-tensorflow-is-using-gpu-acceleration-from-inside-python-shell