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()