运行 多个 GPU 上的 Tensorflow

Question

我查看了该网站，但一如既往地不清楚。任何人都可以描述所有步骤（从一开始）到 运行 GPU 上的任何 tensorflow 程序吗？

Answer 1

来自 Tensorflow 官网： https://www.tensorflow.org/tutorials/using_gpu

# Creates a graph.
c = []
for d in ['/device:GPU:2', '/device:GPU:3']:
  with tf.device(d):
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3])
    b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2])
    c.append(tf.matmul(a, b))
with tf.device('/cpu:0'):
  sum = tf.add_n(c)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(sum))