Breaking News: Grepper is joining You.com. Read the official announcement!

tensorflow configure multiple gpu

Juan David Escobar Escobar answered on April 25, 2022 Popularity 8/10 Helpfulness 2/10

answer tensorflow configure multiple gpu

tensorflow configure multiple gpu

Comment

import tensorflow as tf

c = []
for i, d in enumerate(['/gpu:0', '/gpu:1', '/gpu:2']):
    with tf.device(d):
        a = tf.get_variable(f"a_{i}", [2, 3], initializer=tf.random_uniform_initializer(-1, 1))
        b = tf.get_variable(f"b_{i}", [3, 2], initializer=tf.random_uniform_initializer(-1, 1))
        c.append(tf.matmul(a, b))

with tf.device('/cpu:0'):
    sum = tf.add_n(c)

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

init = tf.global_variables_initializer()
sess.run(init)

print(sess.run(sum))
# [[-0.36499196 -0.07454088]
# [-0.33966339  0.30250686]]

Popularity 8/10 Helpfulness 2/10 Language python

Source: jhui.github.io

Tags: configure multiple-gpu python tensorflow

Link to this answer
Share Copy Link

Contributed on Apr 25 2022

Juan David Escobar Escobar

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

tensorflow configure multiple gpu

Contents

More Related Answers

tensorflow configure multiple gpu

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.