tensorflow tf.device
2018-10-01 本文已影响0人
猴子喜
而一台机器上不同GPU的名称是不同的，第n个GPU在TensorFlow中的名称为/gpu:n。比如第一个GPU的名称为/gpu:0，第二个GPU名称为/gpu:1，以此类推。
这个还属可以把运算放在不同的设备上。
with tf.device('/cpu:0'):
a= tf.constant([1.0, 2.0, 3.0], shape=[3], name='a')
b= tf.constant([1.0, 2.0, 3.0], shape=[3], name='b')
with tf.device('/gpu:1'):
c= a + b
