在多线程中使用tensorRT
仅记录,转自https://www.coder.work/article/4985246import pycuda.autoinit # Create CUDA contextimport pycuda.driver as cudaMain threadwith open(“sample.engine”, “rb”) as f, trt.Runtime(TRT_LOGGER) as runtime:engine = runtime.deserialize_cuda_engine(f.read()
转载
2020-07-24 14:17:19 ·
3269 阅读 ·
4 评论