Python Pipeline 性能测试

May 23, 2022 · View on GitHub

测试环境

测试环境如下表所示:

GPU显存CPU内存
Serving端4x Tesla P4-8GB7611MiBIntel(R) Xeon(R) Gold 5117 CPU @ 2.00GHz 48核216G
Client端4x Tesla P4-8GB7611MiBIntel(R) Xeon(R) Gold 5117 CPU @ 2.00GHz 48核216G

使用单卡GPU,未开启TensorRT。 模型:ResNet_v2_50

性能指标与结论

通过测试,使用 Python Pipeline 模式通过多进程并发,充分利用 GPU 显卡,具有较好的吞吐性能。

测试数据如下:

model_namethread_numbatch_sizeCPU_util(%)GPU_memory(mb)GPU_util(%)qps(samples/s)total countmean(ms)median(ms)80 percent(ms)90 percent(ms)99 percent(ms)total cost(s)each cost(s)
ResNet_v2_50112.2332717.2517.63365886924078735556.42848123899647638.64672851562539.49682617187539.983691406251273.191108398437320.13195347785949720.033540725708008
ResNet_v2_50142.7361728.12253.5074843045352226874.7153921554337874.618164062575.313867187575.605102539062577.8532299804687420.0345804691314720.024930953979492
ResNet_v2_50181.7387725.786959.60582783086999150133.5897119140625132.7503662109375134.968310546875136.470703125140.7903906249999620.13225960731506320.03933620452881
ResNet_v2_501167.0477727.017563.262764681933980252.30162048339844251.8448486328125253.046630859375253.91142578125263.36164062520.23307085037231420.18476152420044
ResNet_v2_501327.5656738.53262.94531468734802440506.8969482421875507.3531494140625510.562353515625511.421240234375536.806892089843720.33511161804199220.276386737823486
ResNet_v2_50214.7656749.482850.40600094376044101039.6335219581528539.534545898437540.45288085937541.137542.94052246093749420.03729677200317420.01696753501892
ResNet_v2_50242.7656744.474483.425583689138242095.3854800269717295.706909179687597.59995117187598.098583984375102.3968090820312520.13770747184753420.03199553489685
ResNet_v2_50282.2656742.89891.3727510505176230174.89108568274457175.0452880859375175.82001953125176.7634033203125178.6406445312500220.1372945308685320.1132071018219
ResNet_v2_502162.265674597.5591285698611124327.16720088835683328.6126708984375329.75185546875330.386962890625336.8639746093749620.33638501167297420.284939169883728
ResNet_v2_502323.2656759.5714100.7076541811633364633.9812698364258637.8568115234375648.103515625650.7439697265625659.221291503906220.33609080314636220.28787398338318
ResNet_v2_50413.1656764.333380.27845081929433161749.5646423075622349.487304687551.553710937552.69340820312555.20756835937499620.14239168167114320.038144528865814
ResNet_v2_50443.3656770.4563136.62061939701394688116.51574919944586121.8629150390625129.8181640625133.384423828125142.6950073242187520.14337229728698720.041599333286285
ResNet_v2_50483.0656770.896158.46554975132275399201.30669079926378210.69775390625228.51748046875236.427294921875252.2482275390623320.14317941665649420.081032752990723
ResNet_v2_504163.2656766.3832156.4935247130092197407.6668608224937423.974609375450.368212890625464.45986328125482.9365820312520.14140844345092820.078101694583893
ResNet_v2_504323.3656772.4791162.01742190796557104785.5079204852765813.0341796875887.107958984375909.6556640625935.333483886718820.54100084304809620.423666059970856
ResNet_v2_50813.5656793.977115.9749228558386233768.558040907814565.4584960937576.1393066406250183.54204101562591.4566601562499820.1509079933166520.028797417879105
ResNet_v2_50844.2656790.0952175.58748591910316889180.7330482920592170.5810546875218.99931640625240.06337890625002254.41375976562520.25201272964477520.084695398807526
ResNet_v2_50882.6656793.8693206.76595246418208526306.52158695119414303.043212890625321.0791015625350.5477294921875400.3245239257812520.35151314735412620.15437400341034
ResNet_v2_508163.2656785.7273205.31850043117367265614.1745522553066552.372314453125775.89169921875802.022607421875902.276318359376120.65084242820739720.345011442899704
ResNet_v2_508325.0656789.8717219.84102737188351461138.45334740207611039.6408691406251364.2897949218751474.67443847656251788.261437988283421.25172019004821820.777225106954575
ResNet_v2_501215.0656789.4762110.008583278478622218108.50048552943953103.015625121.09404296875003137.1392333984375151.8040112304687220.16206312179565420.055511037508648
ResNet_v2_501244.1656777.7619153.7824464757549779309.68895575507463285.585205078125378.07421875413.481640625424.7085351562520.26239061355590820.104551911354065
ResNet_v2_501283.6656772.6977165.36021780846013425571.1991590073529510.995849609375731.9383300781251747.6568359375757.30471679687520.5611727237701420.230452219645183
ResNet_v2_5012161.5656776.2222189.6414991568285252987.7153136238219926.003906251080.991308593751249.49562988281261434.480239257812421.2611691951751720.74245794614156
ResNet_v2_5012322.8656784.25203.8682282817841381811.6402375594431764.27600097656251855.280468752023.568261718752586.803813476562521.6610505580902120.834286351998646
ResNet_v2_501614.8656794.3333116.349277333122342347136.7957122373642135.959716796875144.1568359375146.105517578125175.0570751953124820.17202043533325220.067057371139526
ResNet_v2_5016415.4656783.6364160.59012047270738822393.3079394412447396.446533203125426.272216796875429.777734375564.111936035156220.4744849205017120.206754431128502
ResNet_v2_501686.8656781.0233169.95774070621547437741.5512622684854751.521484375763.199169921875948.80419921875011001.15614257812520.5698192119598420.254074171185493
ResNet_v2_5016163.5656777.8706186.566000815162481332.10079463835681365.27453613281251399.2122558593751432.40373535156251771.437485351562621.2686126232147220.64799252152443
ResNet_v2_5016324.3656783.6371201.12934086381951402419.34001988002232561.092285156252616.0811035156252642.08352050781242883.819741210936622.27422404289245621.169659316539764