linux.md 7.9 KB
Newer Older
F
FlyingQianMM 已提交
1
# Linux平台部署
C
Channingss 已提交
2 3

## 说明
C
Channingss 已提交
4
本文档在 `Linux`平台使用`GCC 4.8.5``GCC 4.9.4`测试过,如果需要使用更高G++版本编译使用,则需要重新编译Paddle预测库,请参考: [从源码编译Paddle预测库](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/advanced_guide/inference_deployment/inference/build_and_install_lib_cn.html#id12)
C
Channingss 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

## 前置条件
* G++ 4.8.2 ~ 4.9.4
* CUDA 9.0 / CUDA 10.0, CUDNN 7+ (仅在使用GPU版本的预测库时需要)
* CMake 3.0+

请确保系统已经安装好上述基本软件,**下面所有示例以工作目录 `/root/projects/`演示**

### Step1: 下载代码

 `git clone https://github.com/PaddlePaddle/PaddleX.git`

**说明**:其中`C++`预测代码在`/root/projects/PaddleX/deploy/cpp` 目录,该目录不依赖任何`PaddleX`下其他目录。


J
jiangjiajun 已提交
20
### Step2: 下载PaddlePaddle C++ 预测库 paddle_inference
C
Channingss 已提交
21

J
jack 已提交
22
PaddlePaddle C++ 预测库针对不同的`CPU``CUDA`,以及是否支持TensorRT,提供了不同的预编译版本,目前PaddleX依赖于Paddle1.8版本,以下提供了多个不同版本的Paddle预测库:
C
Channingss 已提交
23

J
jack 已提交
24
|  版本说明   | 预测库(1.8.2版本)  |
C
Channingss 已提交
25
|  ----  | ----  |
J
jiangjiajun 已提交
26 27 28 29 30 31
| ubuntu14.04_cpu_avx_mkl  | [paddle_inference](https://paddle-inference-lib.bj.bcebos.com/1.8.2-cpu-avx-mkl/fluid_inference.tgz) |
| ubuntu14.04_cpu_avx_openblas  | [paddle_inference](https://paddle-inference-lib.bj.bcebos.com/1.8.2-cpu-avx-openblas/fluid_inference.tgz) |
| ubuntu14.04_cpu_noavx_openblas  | [paddle_inference](https://paddle-inference-lib.bj.bcebos.com/1.8.2-cpu-noavx-openblas/fluid_inference.tgz) |
| ubuntu14.04_cuda9.0_cudnn7_avx_mkl  | [paddle_inference](https://paddle-inference-lib.bj.bcebos.com/1.8.2-gpu-cuda9-cudnn7-avx-mkl/fluid_inference.tgz) |
| ubuntu14.04_cuda10.0_cudnn7_avx_mkl  | [paddle_inference](https://paddle-inference-lib.bj.bcebos.com/1.8.2-gpu-cuda10-cudnn7-avx-mkl/fluid_inference.tgz ) |
| ubuntu14.04_cuda10.1_cudnn7.6_avx_mkl_trt6  | [paddle_inference](https://paddle-inference-lib.bj.bcebos.com/1.8.2-gpu-cuda10.1-cudnn7.6-avx-mkl-trt6%2Ffluid_inference.tgz) |
C
Channingss 已提交
32

C
Channingss 已提交
33
更多和更新的版本,请根据实际情况下载:  [C++预测库下载列表](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/advanced_guide/inference_deployment/inference/build_and_install_lib_cn.html)
C
Channingss 已提交
34 35 36 37 38 39 40 41 42 43 44

下载并解压后`/root/projects/fluid_inference`目录包含内容为:
```
fluid_inference
├── paddle # paddle核心库和头文件
|
├── third_party # 第三方依赖库和头文件
|
└── version.txt # 版本和编译信息
```

C
Channingss 已提交
45
**注意:** 预编译版本除`nv-jetson-cuda10-cudnn7.5-trt5` 以外其它包都是基于`GCC 4.8.5`编译,使用高版本`GCC`可能存在 `ABI`兼容性问题,建议降级或[自行编译预测库](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/advanced_guide/inference_deployment/inference/build_and_install_lib_cn.html#id12)
C
Channingss 已提交
46 47


J
jiangjiajun 已提交
48
### Step3: 编译
C
Channingss 已提交
49 50 51 52

编译`cmake`的命令在`scripts/build.sh`中,请根据实际情况修改主要参数,其主要内容说明如下:
```
# 是否使用GPU(即是否使用 CUDA)
53
WITH_GPU=OFF
54 55
# 使用MKL or openblas
WITH_MKL=ON
C
Channingss 已提交
56 57
# 是否集成 TensorRT(仅WITH_GPU=ON 有效)
WITH_TENSORRT=OFF
58 59 60 61
# TensorRT 的路径,如果需要集成TensorRT,需修改为您实际安装的TensorRT路径
TENSORRT_DIR=/root/projects/TensorRT/
# Paddle 预测库路径, 请修改为您实际安装的预测库路径
PADDLE_DIR=/root/projects/fluid_inference
62 63
# Paddle 的预测库是否使用静态库来编译
# 使用TensorRT时,Paddle的预测库通常为动态库
64
WITH_STATIC_LIB=OFF
C
Channingss 已提交
65
# CUDA 的 lib 路径
66
CUDA_LIB=/usr/local/cuda/lib64
C
Channingss 已提交
67
# CUDNN 的 lib 路径
68
CUDNN_LIB=/usr/local/cuda/lib64
C
Channingss 已提交
69

F
FlyingQianMM 已提交
70 71 72 73 74 75
# 是否加载加密后的模型
WITH_ENCRYPTION=ON
# 加密工具的路径, 如果使用自带预编译版本可不修改
sh $(pwd)/scripts/bootstrap.sh # 下载预编译版本的加密工具
ENCRYPTION_DIR=$(pwd)/paddlex-encryption

76
# OPENCV 路径, 如果使用自带预编译版本可不修改
77
sh $(pwd)/scripts/bootstrap.sh  # 下载预编译版本的opencv
C
Channingss 已提交
78 79 80
OPENCV_DIR=$(pwd)/deps/opencv3gcc4.8/

# 以下无需改动
C
Channingss 已提交
81 82 83 84 85
rm -rf build
mkdir -p build
cd build
cmake .. \
    -DWITH_GPU=${WITH_GPU} \
86
    -DWITH_MKL=${WITH_MKL} \
C
Channingss 已提交
87
    -DWITH_TENSORRT=${WITH_TENSORRT} \
F
FlyingQianMM 已提交
88
    -DWITH_ENCRYPTION=${WITH_ENCRYPTION} \
89
    -DTENSORRT_DIR=${TENSORRT_DIR} \
C
Channingss 已提交
90
    -DPADDLE_DIR=${PADDLE_DIR} \
91
    -DWITH_STATIC_LIB=${WITH_STATIC_LIB} \
C
Channingss 已提交
92 93
    -DCUDA_LIB=${CUDA_LIB} \
    -DCUDNN_LIB=${CUDNN_LIB} \
F
FlyingQianMM 已提交
94
    -DENCRYPTION_DIR=${ENCRYPTION_DIR} \
C
Channingss 已提交
95 96 97
    -DOPENCV_DIR=${OPENCV_DIR}
make
```
C
Channingss 已提交
98
**注意:** linux环境下编译会自动下载OPENCV, PaddleX-Encryption和YAML,如果编译环境无法访问外网,可手动下载:
C
Channingss 已提交
99

C
Channingss 已提交
100 101 102
- [opencv3gcc4.8.tar.bz2](https://paddleseg.bj.bcebos.com/deploy/docker/opencv3gcc4.8.tar.bz2)
- [paddlex-encryption.zip](https://bj.bcebos.com/paddlex/tools/paddlex-encryption.zip)
- [yaml-cpp.zip](https://bj.bcebos.com/paddlex/deploy/deps/yaml-cpp.zip)
C
Channingss 已提交
103

C
Channingss 已提交
104
opencv3gcc4.8.tar.bz2文件下载后解压,然后在script/build.sh中指定`OPENCE_DIR`为解压后的路径。
C
Channingss 已提交
105

C
Channingss 已提交
106
paddlex-encryption.zip文件下载后解压,然后在script/build.sh中指定`ENCRYPTION_DIR`为解压后的路径。
C
Channingss 已提交
107

C
Channingss 已提交
108
yaml-cpp.zip文件下载后无需解压,在cmake/yaml.cmake中将`URL https://bj.bcebos.com/paddlex/deploy/deps/yaml-cpp.zip` 中的网址,改为下载文件的路径。
C
Channingss 已提交
109 110 111 112 113 114

修改脚本设置好主要参数后,执行`build`脚本:
 ```shell
 sh ./scripts/build.sh
 ```

J
jiangjiajun 已提交
115
### Step4: 预测及可视化
F
FlyingQianMM 已提交
116

J
jiangjiajun 已提交
117
**在加载模型前,请检查你的模型目录中文件应该包括`model.yml`、`__model__`和`__params__`三个文件。如若不满足这个条件,请参考[模型导出为Inference文档](../../export_model.md)将模型导出为部署格式。**  
J
Jason 已提交
118

119
编译成功后,预测demo的可执行程序分别为`build/demo/detector``build/demo/classifier``build/demo/segmenter`,用户可根据自己的模型类型选择,其主要命令参数说明如下:
C
Channingss 已提交
120 121 122 123 124 125 126

|  参数   | 说明  |
|  ----  | ----  |
| model_dir  | 导出的预测模型所在路径 |
| image  | 要预测的图片文件路径 |
| image_list  | 按行存储图片路径的.txt文件 |
| use_gpu  | 是否使用 GPU 预测, 支持值为0或1(默认值为0) |
127
| use_trt  | 是否使用 TensorRT 预测, 支持值为0或1(默认值为0) |
C
Channingss 已提交
128
| gpu_id  | GPU 设备ID, 默认值为0 |
J
Jason 已提交
129
| save_dir | 保存可视化结果的路径, 默认值为"output",**classfier无该参数** |
130 131 132
| key | 加密过程中产生的密钥信息,默认值为""表示加载的是未加密的模型 |
| batch_size | 预测的批量大小,默认为1 |
| thread_num | 预测的线程数,默认为cpu处理器个数 |
J
jiangjiajun 已提交
133
| use_ir_optim | 是否使用图优化策略,支持值为0或1(默认值为1,图像分割默认值为0)|
C
Channingss 已提交
134

C
Channingss 已提交
135 136
## 样例

J
jiangjiajun 已提交
137
可使用[小度熊识别模型](../../export_model.md)中导出的`inference_model`和测试图片进行预测,导出到/root/projects,模型路径为/root/projects/inference_model。
C
Channingss 已提交
138

F
FlyingQianMM 已提交
139
> 关于预测速度的说明:加载模型后前几张图片的预测速度会较慢,这是因为运行启动时涉及到内存显存初始化等步骤,通常在预测20-30张图片后模型的预测速度达到稳定。
J
jiangjiajun 已提交
140

C
Channingss 已提交
141 142
`样例一`

143
不使用`GPU`测试图片 `/root/projects/images/xiaoduxiong.jpeg`  
C
Channingss 已提交
144 145

```shell
146
./build/demo/detector --model_dir=/root/projects/inference_model --image=/root/projects/images/xiaoduxiong.jpeg --save_dir=output
C
Channingss 已提交
147 148 149 150 151 152
```
图片文件`可视化预测结果`会保存在`save_dir`参数设置的目录下。


`样例二`:

153
使用`GPU`预测多个图片`/root/projects/image_list.txt`,image_list.txt内容的格式如下:
C
Channingss 已提交
154
```
155 156
/root/projects/images/xiaoduxiong1.jpeg
/root/projects/images/xiaoduxiong2.jpeg
C
Channingss 已提交
157
...
158
/root/projects/images/xiaoduxiongn.jpeg
C
Channingss 已提交
159 160
```
```shell
161
./build/demo/detector --model_dir=/root/projects/inference_model --image_list=/root/projects/images_list.txt --use_gpu=1 --save_dir=output --batch_size=2 --thread_num=2
C
Channingss 已提交
162 163
```
图片文件`可视化预测结果`会保存在`save_dir`参数设置的目录下。