如何在张量流中使用多个GPU？

Question

代码如下：

TF_SESSION_CONFIG = tf.compat.v1.ConfigProto(
    gpu_options=tf.compat.v1.GPUOptions(allow_growth=True),
    log_device_placement=False,
    device_count={"GPU": 1}
    )

上面的代码仅使用机器的一个GPU。我尝试使用另一个GPU，并将代码更改为：

TF_SESSION_CONFIG = tf.compat.v1.ConfigProto(
    gpu_options=tf.compat.v1.GPUOptions(allow_growth=True),
    log_device_placement=False,
    device_count={"GPU": 0, "GPU": 1}
    )

这种方法也不起作用。那么如何使用两个GPU来完成任务？