tensorflow为什么只把数据放在一张gpu

代码中设置了4张gpu
但是训练过程只用了第一张gpu
求xdm解答

• 这篇文章：tensorflow-gpu的一个问题 也许有你想要的答案，你可以看看

不知道你的代码是什么样的，试试下面的代码：

strategy = tf.distribute.MirroredStrategy()
with strategy.scope():
  inputs = tf.keras.layers.Input(shape=(1,))
  predictions = tf.keras.layers.Dense(1)(inputs)
  model = tf.keras.models.Model(inputs=inputs, outputs=predictions)
  model.compile(loss='mse',
                optimizer=tf.keras.optimizers.SGD(learning_rate=0.2))
model.fit( ... )

如果还不行，你可以把多卡相关的代码贴一下