Question

如何构建连接到现有TF服务器集的TensorFlow会话？

在distributed TensorFlow指南之后，我启动了一堆TensorFlow服务器，这些服务器连接成一个集群。

我现在想开始一个可以连接到这些TF服务器并为其分配操作的会话。我假设我只需要在tf会话的构造函数中指定一个合适的目标;例如

with tf.Session(
    target, config=tf.ConfigProto(log_device_placement=True)) as sess:

但是我不清楚如何构建指向任何现有TF服务器集群的目标对象。这些文档仅显示如何通过调用server.target从服务器获取集群规范。

我是否需要启动另一台服务器才能构建与现有服务器通信的客户端？

我想远程连接到TF群集。我的TF服务器正在GCE VM上运行。我想从我的本地机器连接并分配操作。这可能吗？

Answer 1

对于目标，您只需使用

形式的字符串即可

grpc://<host>:<port>

其中host和port是组成群集的TF服务器之一的主机名（或IP地址）和端口。它并不重要，因为op执行的位置取决于构造图形时分配给它的设备。