2022年3月24日
triton使用踩坑
摘要： 1、启动服务时，默认会加载tensorflow1的backend so库。如果想要使用tensorflow2，则在启动命令中添加： --backend-config="tensorflow,version=2" 则会自动加载镜像中/opt/tritonserver/backend/tensorflo 阅读全文
posted @ 2022-03-24 14:48 鸭子船长 阅读(432) 评论(0) 推荐(0)
查看Tensorflow SavedModel模型信息（转）
摘要： 当我们拿到别人提供的pb模型时，我们很可能不清楚inputs所需的参数，可以使用以下方法来查看模型信息： 找到你的tensorflow安装位置，依次定位到以下目录:tensorflow_core/python/tools执行如下命令：python saved_model_cli.py show -- 阅读全文
posted @ 2022-03-24 14:43 鸭子船长 阅读(308) 评论(0) 推荐(0)