使用perf_analyzer和model-analyzer测试tritonserver的模型性能超详细完整版

修炼之路2023-09-06 8:36

导读

当我们在使用tritonserver部署模型之后，通常需要测试一下模型的服务QPS的能力，也就是1s我们的模型能处理多少的请求，也被称为吞吐量。

测试tritonserver模型服务的QPS通常有两种方法，一种是使用perf_analyzer 来测试，另一种是通过model-analyzer来获取更为详细的模型服务启动的参数，使得模型的QPS达到最大，下面我们将分别来介绍如何通过这两种工具来测试tritonserver模型服务的吞吐量。

环境准备

使用docker镜像来进行测试非常的方便，一般对应版本的tritonserver都会有一个tritonserver-sdk的版本，例如21.10需要准备的镜像如下

nvcr.io/nvidia/tritonserver:21.10-py3

nvcr.io/nvidia/tritonserver:21.10-py3-sdk

nvcr.io/nvidia/tensorrt:21.10-py3

  • tritonserver：tritonserver模型启动服务所需要的镜像
  • tritonserver-py3-sdk：包含perf_analyzer 和model-analyzer可以用于做性能测试的镜像
  • tensorrt：包含与tritonserver镜像中一样的tensorrt版本，该镜像主要用来将onnx模型转换为engine用于tritonserver启动模型服务时所需要的模型文件

注：不同版本的版本的镜像所需要的nvidia-driver的版本不一样，21.10所需要的nvida-driver版本为>=470.82(cuda11.4)，如果低于该版本时，在启动镜像的时候会提示不兼容。在准备上面镜像的时候tritonserver和tensorrt版本一定要一致，因为不同的版本依赖库以及tensorrt的版本可能会不一样，导致模型可能无法通用

perf_analyzer

  1. 启动tritonserver模型服务
bash 复制代码 
docker run --gpus all --rm -p8000:8000 -p8001:8001 -p8002:8002 -v triton_models:/models nvcr.io/nvidia/tritonserver:21.10-py3 tritonserver --model-repository=/models
  • --gpus：表示启动tritonserver服务是否使用gpu
  • --rm：当退出容器的时候会自动删除容器
  • -p：用于容器内和宿主机之间的端口映射
  • --model-repository：指定tritonserver运行的时候所需要的模型文件

    启动之后输出上面的信息就表示tritonserver模型服务启动成功
  1. 运行tritonserver-sdk镜像
bash 复制代码 
docker run -it --rm --net=host nvcr.io/nvidia/tritonserver:21.10-py3-sdk

因为这里我们只是使用了perf_analyzer性能测试工具，所以其他的启动参数可以不用管

  1. 性能测试
bash 复制代码 
perf_analyzer -m resnet34 --percentile=95 --concurrency-range 100
  • -m：指定测试的模型
  • --percentile：以百分位数表示置信度值用于确定测量结果是否稳定
  • --concurrency-range：指定并发量参数，还可以使用，100:120:10表示并发量在100到120之间以10作为间隔测试所有并发量情况下的吞吐量
  • -b：指定请求的batch-size，默认为1
  • -i：指定请求所使用的协议，参数可以为http或grpc

其他更多的参数可以通过perf_analyzer -h查看

执行成功之后会输出模型所能承受的吞吐量以及时延

*** Measurement Settings ***

Batch size: 1

Using "time_windows" mode for stabilization

Measurement window: 5000 msec

Using synchronous calls for inference

Stabilizing using p95 latency

Request concurrency: 100

Client:

Request count: 11155

Throughput: 2231 infer/sec

p50 latency: 43684 usec

p90 latency: 61824 usec

p95 latency: 67900 usec

p99 latency: 76275 usec

Avg HTTP time: 45183 usec (send/recv 161 usec + response wait 45022 usec)

Server:

Inference count: 13235

Execution count: 722

Successful request count: 722

Avg request latency: 29032 usec (overhead 629 usec + queue 11130 usec + compute input 7317 usec + compute infer 9889 usec + compute output 67 usec)

Inferences/Second vs. Client p95 Batch Latency

Concurrency: 100, throughput: 2231 infer/sec, latency 67900 usec

在使用http协议进行测试的时候，可能会存在比较大的波动，小伙伴们可以多测试几组来取平均值，grpc协议相对来说更加稳定一些。

我们最重要的是关注最后一行的输出信息，表示在100并发的请求下，tritonserver的吞吐量可以达到2231请求/s，时延是67900微秒。可以看出tritonserver模型服务的推理能力还是很强的，显卡是RTX3090，在只使用了一个count的情况下ResNet34可以达到2200FPS

model-analyzer

  1. 启动tritonserver-sdk
bash 复制代码 
docker run --gpus 1 --shm-size=1g --ulimit memlock=-1 --ulimit stack=67108864 -ti -v /var/run/docker.sock:/var/run/docker.sock --net host --privileged --rm -v /home/***/triton_models:/models -v /data/reports:/data/reports nvcr.io/nvidia/tritonserver:21.10-py3-sdk bash

这里需要注意的是，我们在这里不需要再去启动tritonserver模型服务了，但是启动的参数多了很多，上面这些参数都是必要的

  • v指定docker.sock是让容器能够启动tritonser模型服务容器
  • /data/reports:/data/reports：指定模型测试报告存储的路径，这里容器内外的路径需要保持一致，否则后面再使用model-analyzer的时候会报错
  1. 创建tritonserver可执行文件
bash 复制代码 
touch /opt/tritonserver
chmod +x /opt/tritonserver
  1. 启动model-analyzer测试
bash 复制代码 
model-analyzer profile --model-repository /models --profile-models resnet34 --run-config-search-max-concurrency 2 --run-config-search-max-instance-count 2 --gpus '0' --triton-launch-mode=docker --output-model-repository=/data/reports/resnet34 --triton-server-path=/opt/tritonserver --override-output-model-repository
  • profile：
  • model-repository：指定模型保存的目录，这里可以是包含多个模型的目录
  • profile-models ：指定测试模型的名字
  • run-config-search-max-concurrency：最大的并发数
  • run-config-search-max-instance-count：最大的count也就是模型实例数
  • triton-launch-mode：docker或local，如果是local的时候需要容器内安装tritonserver，这里使用的是docker
  • output-model-repository：测试报告保存的目录
  • override-output-model-repository：每次测试的时候都会覆盖之前的测试报告

为了加快测试的速度，这里我将count设置为2，concurrency 也设置为2，在实际使用的过程中小伙伴们可以设置一个范围，使用min和max来进行控制，而不要直接固定住这些参数。因为我们最终需要的是得到count、concurrency、batch-size这些参数的最优值。

注：需要注意triton-model-analyzer的版本，不同的版本指定的参数会不一样，这里triton-model-analyzer==1.9.0，小伙伴们可以通过model-analyzer -h来查看参数。

执行成功之后我们会得到一个ckpt的文件

  1. 导出模型的性能报告
bash 复制代码 
mkdir analysis_results
model-analyzer analyze --analysis-models resnet34 -e analysis_results
  1. 查看结果

拷贝报告，将容器内的报告拷贝到宿主机上，通过docker ps -a来查看容器的id

bash 复制代码 
docker cp 5610298c2c93:/workspace/analysis_results ./

在/analysis_results/reports/summaries/resnet34目录下有一个result_summary.pdf文件里面记录了模型的性能参数

这里吞吐量低的主要原因是因为我们把最大的并发数设置为2了，大家在使用的时候可以设置大一点，参数搜索的范围越大所需要的时间就会越长

相关错误以及解决办法

  1. 启动tritonserver服务失败

I0903 13:53:33.877351 1 server.cc:233] Waiting for in-flight requests to complete.

I0903 13:53:33.877375 1 server.cc:248] Timeout 30: Found 0 live models and 0 in-flight non-inference requests

error: creating server: Internal - failed to load all models

模型服务启动失败时，注意看前面的错误信息，前面会有显示每个模型的加载状态以及加载失败的原因

上面错误的原因是因为在转换启动模型文件的时候，所使用的tensorrt版本与容器内tensorrt的版本不一致导致加载失败，通过使用与tritonserver版本一致的tensorrt进行将onnx模型重新转换一下即可。

  1. Failed to set the value for field "triton_server_path"

2023-09-03 14:45:35.892 ERROR[entrypoint.py:214] Model Analyzer encountered an error: Failed to set the value for field "triton_server_path". Error: Either the binary 'tritonserver' is not on the PATH, or Model Analyzer does not have permissions to execute os.stat on this path.

bash 复制代码 
touch /opt/tritonserver
chmod +x /opt/tritonserver
  1. OSError: [Errno 16] Device or resource busy: '/data/reports'
    在执行model-analyzer的时候提示resource busy，详细错误信息如下

2023-09-04 02:52:07.192 ERROR[entrypoint.py:214] Model Analyzer encountered an error: Failed to set the value for field "triton_server_path". Error: Either the binary '/opt/tritonserver' is not on the PATH, or Model Analyzer does not have permissions to execute os.stat on this path.

root@zhouwen3090:/workspace# touch /opt/tritonserver

root@zhouwen3090:/workspace# chmod +x /opt/tritonserver

root@zhouwen3090:/workspace# model-analyzer profile --model-repository /models --profile-models resnet34 --run-config-search-max-concurrency 1 --run-config-search-max-instance-count 1 --gpus '0' --triton-launch-mode=docker --output-model-repository=/data/models/reports --triton-server-path=/opt/tritonserver --override-output-model-repository

2023-09-04 02:52:22.668 INFO[gpu_device_factory.py:50] Initiliazing GPUDevice handles...

2023-09-04 02:52:23.764 INFO[gpu_device_factory.py:246] Using GPU 0 NVIDIA GeForce RTX 3090 with UUID GPU-82eb2e4d-c579-7908-39c9-92201438f73c

Traceback (most recent call last):

File "/usr/local/bin/model-analyzer", line 8, in

sys.exit(main())

File "/usr/local/lib/python3.8/dist-packages/model_analyzer/entrypoint.py", line 289, in main

create_output_model_repository(config)

File "/usr/local/lib/python3.8/dist-packages/model_analyzer/entrypoint.py", line 249, in create_output_model_repository

os.mkdir(config.output_model_repository_path)

FileNotFoundError: [Errno 2] No such file or directory: '/data/models/reports'

root@zhouwen3090:/workspace# model-analyzer profile --model-repository /models --profile-models resnet34 --run-config-search-max-concurrency 1 --run-config-search-max-instance-count 1 --gpus '0' --triton-launch-mode=docker --output-model-repository=/data/reports --triton-server-path=/opt/tritonserver --override-output-model-repository

2023-09-04 02:53:38.665 INFO[gpu_device_factory.py:50] Initiliazing GPUDevice handles...

2023-09-04 02:53:39.750 INFO[gpu_device_factory.py:246] Using GPU 0 NVIDIA GeForce RTX 3090 with UUID GPU-82eb2e4d-c579-7908-39c9-92201438f73c

Traceback (most recent call last):

File "/usr/local/lib/python3.8/dist-packages/model_analyzer/entrypoint.py", line 249, in create_output_model_repository

os.mkdir(config.output_model_repository_path)

FileExistsError: [Errno 17] File exists: '/data/reports'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):

File "/usr/local/bin/model-analyzer", line 8, in

sys.exit(main())

File "/usr/local/lib/python3.8/dist-packages/model_analyzer/entrypoint.py", line 289, in main

create_output_model_repository(config)

File "/usr/local/lib/python3.8/dist-packages/model_analyzer/entrypoint.py", line 258, in create_output_model_repository

shutil.rmtree(config.output_model_repository_path)

File "/usr/lib/python3.8/shutil.py", line 722, in rmtree

onerror(os.rmdir, path, sys.exc_info())

File "/usr/lib/python3.8/shutil.py", line 720, in rmtree

os.rmdir(path)

OSError: [Errno 16] Device or resource busy: '/data/reports'

只需要修改一下--output-model-repository=/data/reports指定的路径即可，在后面增加一个模型的名称修改为--output-model-repository=/data/reports/resnet34

  1. /usr/bin/wkhtmltopdf: error while loading shared libraries: libQt5Core.so.5: cannot open shared object file: No such file or directory
bash 复制代码 
strip --remove-section=.note.ABI-tag /usr/lib/x86_64-linux-gnu/libQt5Core.so.5

参考

  1. https://github.com/triton-inference-server/client/blob/main/src/c%2B%2B/perf_analyzer/docs/cli.md
  2. https://github.com/triton-inference-server/model_analyzer/tree/main/docs
  3. https://github.com/triton-inference-server/model_analyzer/blob/main/docs/quick_start.md
  4. https://github.com/triton-inference-server/model_analyzer/blob/main/docs/install.md
  5. https://github.com/triton-inference-server/model_analyzer/issues/120
热门推荐
01【音视频 | opus】opus编解码库(opus-1.4)详细介绍以及使用——附带解码示例代码02PyTorch机器学习实现液态神经网络03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04ICLR24大模型提示(1/11) | BadChain：大型语言模型的后门思维链提示05Coze扣子平台完整体验和实践（附国内和国际版对比）06(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明07〔AI 绘画〕Stable Diffusion 之 解决绘制多人或面部很小的人物时面部崩坏问题 篇08玄机平台应急响应—webshell查杀09RAG 实践- Ollama+RagFlow 部署本地知识库10国内外大模型汇总：Open AI大模型、Google大模型、Microsoft大模型、文心一言大模型、通义千问大模型、字节豆包大模型、智普清言大模型