Skip to content

Commit 2b1ea65

Browse files
committed
inference reademe
1 parent 61a4694 commit 2b1ea65

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

doc/inference.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,13 @@ python -u ../../../tools/static_trainer.py -m config.yaml # 全量数据运行co
2626
## 将保存的模型使用Inference预测库进行服务端部署
2727
paddlerec提供tools/paddle_infer.py脚本,供您方便的使用inference预测库高效的对模型进行预测。
2828

29+
需要安装的库:
30+
```bash
31+
pip install pynvml
32+
pip install psutil
33+
pip install GPUtil
34+
```
35+
2936
1. 启动paddle_infer.py脚本的参数:
3037

3138
| 名称 | 类型 | 取值 | 是否必须 | 作用描述 |

0 commit comments

Comments
 (0)