GRPC for torchserve

该博客介绍了如何通过GRPC接口与torchserve结合,完成模型的注册、推理和注销操作。首先,需要安装torchserve及相关依赖,然后创建并启动torchserve服务。接着,利用proto文件生成Python GRPC客户端代码。最后,通过客户端脚本进行densenet161模型的注册、推理和注销演示。
摘要由CSDN通过智能技术生成

GRPC for torchserve

依赖安装

torchserve相关
pip install torchserve torch-model-archiver torch-workflow-archiver
GRPC相关
pip install -U grpcio protobuf grpcio-tools

启动torchserve

mkdir model_store
torchserve --start --ts-config ./config.properties  --model-store model_store --ncs

使用 proto 文件生成 python gRPC客户端

python -m grpc_tools.protoc --proto_path=proto/ --python_out=ts_scripts --grpc_python_out=ts_scripts proto/inference.proto proto/management.proto

densenet161模型注册

python ts_scripts/torchserve_grpc_client.py register densenet161

densenet161模型推理

python ts_scripts/torchserve_grpc_client.py infer densenet161 examples/image_classifier/kitten.jpg

densenet161模型注销

python ts_scripts/torchserve_grpc_client.py unregister densenet161
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

发呆的比目鱼

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值