如何查看tensorflow是GPU还是CPU版本
运行下面python代码,查看输出结果sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))sess.close() 如果输出结果里面有显卡型号信息(如下图),则安装的tensorflow为GPU版,否则为CPU版
参考:https://stackoverflow.com/questions/38009682/how-to-tell-if-tensorflow-is-using-gpu-acceleration-from-inside-python-shell
页:
[1]