Python PyTorch EtcdServer用法及代码示例

本文简要介绍python语言中 torch.distributed.elastic.rendezvous.etcd_server.EtcdServer 的用法。

用法:

class torch.distributed.elastic.rendezvous.etcd_server.EtcdServer(data_dir=None)

参数：

etcd_binary_path-etcd 服务器二进制文件的路径(参见上面的备用路径)

注意

在 etcd 服务器 v3.4.3 上测试

在随机空闲端口上启动和停止本地独立 etcd 服务器。对单节点很有用，multi-worker 启动或测试，其中 sidecar etcd 服务器比单独设置 etcd 服务器更方便。

此类注册一个终止处理程序以在退出时关闭 etcd 子进程。此终止处理程序不能替代调用 stop() 方法。

以下回退机制用于查找 etcd 二进制文件：

1. 使用环境变量TORCHELASTIC_ETCD_BINARY_PATH

2. 如果存在，则使用 <this file root>/bin/etcd

3. 使用 PATH 中的 etcd

用法

server = EtcdServer("/usr/bin/etcd", 2379, "/tmp/default.etcd")
server.start()
client = server.get_client()
# use client
server.stop()