NVIDIA / trt-samples-for-hackathon-cn

Simple samples for TensorRT programming
Apache License 2.0
1.48k stars 337 forks source link

关于09-advance里面的multiSteam,关于run2函数,如何传入data数据进行推理? #51

Closed mingwz closed 1 year ago

wili-65535 commented 2 years ago

I'm sorry for remaining this problem in the sample code Orz.

You can refer to the sample code in 09-Advance/StreamAndAsync to solve this problem (I will fix it in the same way later).

In general, we need to take out the pointer of the allocated page-locked memory buffer, pack it into a numpy array, and then use this numpy arrays to write and read the data in Python.

mingwz commented 2 years ago

感谢大佬的提醒,我悟了 @wili-65535