无涯教程-TensorFlow - 分布式计算

本章将重点介绍如何开始使用分布式TensorFlow，目的是帮助开发人员了解重复出现的基本分布式TF概念，如TF服务器。无涯教程将使用Jupyter Notebook分布式TensorFlow。

第1步 - 导入分布式计算必需的必要模块-

import tensorflow as tf

第2步 - 创建具有一个节点的TensorFlow集群，让此节点负责一个名称为" worker"的作业，该作业将在localhost:2222进行一次操作。

cluster_spec=tf.train.ClusterSpec({worker : [localhost:2222]})
server=tf.train.Server(cluster_spec)
server.target

上面的脚本生成以下输出-

grpc://localhost:2222
The server is currently running.

第3步 - 可以通过执行以下命令来计算具有相应会话的服务器配置-

server.server_def

上面的命令生成以下输出-

cluster {job {name: "worker"tasks {value: "localhost:2222"}}
}
job_name: "worker"
protocol: "grpc"

第4步 - 在执行引擎为服务器的情况下启动TensorFlow会话，使用TensorFlow创建本地服务器，然后使用 lsof 查找服务器的位置。

sess=tf.Session(target=server.target)
server=tf.train.Server.create_local_server()

第5步 - 查看此会话中可用的设备并关闭相应的会话。

devices=sess.list_devices()
for d in devices:print(d.name)
sess.close()

上面的命令生成以下输出-

/job:worker/replica:0/task:0/device:CPU:0

TensorFlow - 分布式计算 - 无涯教程网无涯教程网提供本章将重点介绍如何开始使用分布式TensorFlow ，目的是帮助开发人员了解重复出现的基...https://www.learnfk.com/tensorflow/tensorflow-distributed-computing.html