使用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 并行发送多个样本,您需要有多个连接。以下代码示例使用asyncio并aiohttp展示了实现此目的的一种方法: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)