使用ray.serve传递多个对象以进行并行执行

我正在关注ray 文档中的本教程

具体到这部分:

client = serve.start()
config = {"num_replicas": 3}
client.create_backend("tf:v1", TFMnistModel, TRAINED_MODEL_PATH,config=config)
client.create_endpoint("tf_classifier", backend="tf:v1", route="/mnist")

下面将单个样本发送到后端

sample_data= np.random.randn(28 * 28).tolist()
resp = requests.get(
    "http://localhost:8000/mnist",
    json={"array": sample_data})

如何同时发送多个样本,以便利用所有内核并行执行它们?例如,使用以下方法创建的 100 个 MNIST 样本

# 100 MNIST sample 28x28
sample_data = np.random.randn(100 * 28 * 28).reshape((100, 28, 28))

回答

requests.get()呼叫阻塞,所以你说得对,我们不应该只是把它100次在for循环中。

要通过 HTTP 并行发送多个样本,您需要有多个连接。以下代码示例使用asyncioaiohttp展示了实现此目的的一种方法:https : //gist.github.com/architkulkarni/0bd0a92c3195c58ec460a5a0e5eb0e88#file-benchmark-py(您需要编辑url并添加 JSON 输入session.get()以匹配您的示例)

另一种方法是跳过 HTTP 并使用 Ray Serve 的ServeHandle API从 Python 执行此操作。

handle = client.get_handle("tf_classifier")
futures = [handle.remote({"array": np.random.randn(28 * 28)}) for i in range(100)]
results = ray.get(futures)


以上是使用ray.serve传递多个对象以进行并行执行的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>