triton使用踩坑
1、启动服务时,默认会加载tensorflow1的backend so库。如果想要使用tensorflow2,则在启动命令中添加:
--backend-config="tensorflow,version=2"
则会自动加载镜像中/opt/tritonserver/backend/tensorflow2中的backend,当模型指定platform为tensorflow时,会自动使用名称为tensorflow的backend,此时即为tensorflow2.
2、如果想要使用tensorflow saved_model,模型配置方式如下:
对于graphdef模型,triton的默认文件名为model.graphdef。对于savedmodel,会默认在1(version)目录下找到model.savedmodel目录,然后在找对应模型及参数文件
models ├── resnet │ ├── 1 │ │ └── model.savedmodel │ │ ├── saved_model.pb │ │ └── variables │ │ ├── variables.data-00000-of-00001 │ │ └── variables.index │ └── config.pbtxt └── simple ├── 1 │ └── model.graphdef └── config.pbtxt
联系方式:emhhbmdfbGlhbmcxOTkxQDEyNi5jb20=
分类:
Tensorflow
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· DeepSeek 开源周回顾「GitHub 热点速览」
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了