AIcomes 发表于 2019-5-13 14:52:43

如何查看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]
查看完整版本: 如何查看tensorflow是GPU还是CPU版本