yolov5 Tensorrt加速 zed取图像碰到的问题

原程序的功能是将samples文件夹里的照片一次性进行识别。想改变程序，用zed拍照，达到实时显示识别的目的，但是加了while（1）以及zed grab后，报错：
[08/04/2020-16:01:13] [E] [TRT] …/rtSafe/safeContext.cpp (133) - Cudnn Error in configure: 7 (CUDNN_STATUS_MAPPING_ERROR)
[08/04/2020-16:01:13] [E] [TRT] FAILED_EXECUTION: std::exception
^C

问题所在，加载模型的时候 指定了GPU
cudaSetDevice(DEVICE)
而后面进行推理（do inference）的时候没有指定GPU。造成推理程序无法找到GPU

解决办法：
第一种办法：注释掉加载模型前的GPU，这样两个板块都没指定GPU，就都会选默认的GPU。
第二种办法：在推理前也指定相同的GPU，cudaSetDevice(DEVICE)。