在 Tensorflow 代码中指定 gpu: /gpu:0 is always working?

Question

我的工作站有 3 张显卡，其中一张是 Quadro K620，另外两张是 Titan X。现在我想运行我的 tensorflow 代码在一张显卡上，所以我可以让其他人闲着去做另一项任务。

然而，无论设置tf.device('/gpu:0')还是tf.device('/gpu:1')，我发现1st Titan X显卡一直在工作，我不知道为什么。

import argparse
import os
import time
import tensorflow as tf
import numpy as np
import cv2

from Dataset import Dataset
from Net import Net

FLAGS = None

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('--foldername', type=str, default='./data-large/')
    parser.add_argument('--batch_size', type=int, default=100)
    parser.add_argument('--num_epoches', type=int, default=100)
    parser.add_argument('--learning_rate', type=float, default=0.5)

    FLAGS = parser.parse_args()
    net = Net(FLAGS.batch_size, FLAGS.learning_rate)

    with tf.Graph().as_default():
        # Dataset is a class for encapsulate the input pipeline
        dataset = Dataset(foldername=FLAGS.foldername,
                              batch_size=FLAGS.batch_size,
                              num_epoches=FLAGS.num_epoches)

        images, labels = dataset.samples_train

        ## The following code defines the network and train
        with tf.device('/gpu:0'): # <==== THIS LINE
            logits = net.inference(images)
            loss = net.loss(logits, labels)
            train_op = net.training(loss)

            init_op = tf.group(tf.initialize_all_variables(), tf.initialize_local_variables())
            sess = tf.Session()
            sess.run(init_op)
            coord = tf.train.Coordinator()
            threads = tf.train.start_queue_runners(sess=sess, coord=coord)
            start_time = time.time()
            try:
                step = 0
                while not coord.should_stop():
                    _, loss_value = sess.run([train_op, loss])
                    step = step + 1
                    if step % 100 == 0:
                        format_str = ('step %d, loss = %.2f, time: %.2f seconds')
                        print(format_str % (step, loss_value, (time.time() - start_time)))
                        start_time = time.time()
            except tf.errors.OutOfRangeError:
                print('done')
            finally:
                coord.request_stop()

            coord.join(threads)
            sess.close()

关于行“<=== THIS LINE：”

如果我设置 tf.device('/gpu:0')，显示器显示：

|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Quadro K620         Off  | 0000:03:00.0      On |                  N/A |
| 34%   45C    P0     2W /  30W |    404MiB /  1993MiB |      5%      Default |
+-------------------------------+----------------------+----------------------+
|   1  GeForce GTX TIT...  Off  | 0000:04:00.0     Off |                  N/A |
| 22%   39C    P2   100W / 250W |  11691MiB / 12206MiB |      8%      Default |
+-------------------------------+----------------------+----------------------+
|   2  GeForce GTX TIT...  Off  | 0000:81:00.0     Off |                  N/A |
| 22%   43C    P2    71W / 250W |    111MiB / 12206MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

显示第 1 张 Titan X 卡正在运行。

如果我设置 tf.device('/gpu:1')，显示器显示：

|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Quadro K620         Off  | 0000:03:00.0      On |                  N/A |
| 34%   45C    P0     2W /  30W |    411MiB /  1993MiB |      3%      Default |
+-------------------------------+----------------------+----------------------+
|   1  GeForce GTX TIT...  Off  | 0000:04:00.0     Off |                  N/A |
| 22%   52C    P2    73W / 250W |  11628MiB / 12206MiB |     12%      Default |
+-------------------------------+----------------------+----------------------+
|   2  GeForce GTX TIT...  Off  | 0000:81:00.0     Off |                  N/A |
| 22%   42C    P2    71W / 250W |  11628MiB / 12206MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

显示两张 Titan X 卡在工作，而不是第二张 Titan X 单独工作。

那么这背后的任何原因以及如何指定我希望我的程序进入运行的 gpu？

Answer 1

只是一个猜测，但是 tf.train.Optimizer object (which I expect is created in net.training(loss)) when you call minimize() 的默认行为是 colocate_gradients_with_ops=False。这可能会导致反向传播操作被放置在默认设备上，即 /gpu:0.

要确定是否发生这种情况，您可以遍历 sess.graph_def 并查找在 NodeDef.device 字段中具有 /gpu:0 或具有空 [=18] 的节点=] 字段（在这种情况下，它们将默认放置在 /gpu:0 上）。

检查正在使用哪些设备的另一个选项是在运行您的步骤时使用。这显示了 TensorFlow 实际使用的设备（而不是 sess.graph_def 中您的程序请求的设备），并且应该准确显示 /gpu:0.[=25= 上的运行节点。 ]

在 Tensorflow 代码中指定 gpu: /gpu:0 is always working?

Specify gpu in Tensorflow code: /gpu:0 is always working?

multi-gpu

tensorflow