如何在不启动服务器的情况下构建远程 TenosrFlow 会话？

Question

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

按照distributed TensorFlow 指南，我启动了一堆 TensorFlow 服务器，它们连接起来形成一个集群。

我现在想启动一个可以连接到这些 TF 服务器并为它们分配操作的会话。我假设我只需要在 tf session 的构造函数中指定一个适当的目标；例如像

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

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

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

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

Answer 1

对于目标，您可以只使用形式的字符串

grpc://<host>:<port>

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