windows下使用tensorflow指定gpu运行程序

查看CPU/GPU信息:

输入以下命令:

import os from tensorflow.python.client import device_lib os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99" if __name__ == "__main__": print(device_lib.list_local_devices())

我们可以看到:

 

我们可以使用的是: "/device:CPU:0","/device:GPU:0"

---

使用指定GPU运行程序

如果想要看程序是GPU还是CPU跑的，要到终端上去运行python。我的spyder看不到Device Mapping（不知道你们的可以不可以）

输入以下这段代码：

import tensorflow as tf with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)) as sess: sess.run(init) with tf.device('/device:GPU:0'): for _ in range(3): print(sess.run(state))

输入代码后：

 

回车之后：看到Device Mapping信息了

---

查看tensorflow日志：

在代码中加入以下代码，调用这个函数，就会自动在在当前代码所在位置生成日志文件夹。

 

def Log(): input1 = tf.constant([1.0, 2.0, 3.0], name="input1") input2 = tf.Variable(tf.random_uniform([3], name="input2")) output = tf.add_n([input1, input2], name="add") writer = tf.summary.FileWriter("./log", tf.get_default_graph()) writer.close()

然后在命令行下，进入代码所在的文件夹：

输入以下代码：

tensorboard --logdir=log

然后出现以下结果：

打开浏览器，进入这个网址。

 

---