错误[TensorRT] ERROR: cuda/reformat.cu (773) - Cuda Error in NCHWToNCHHW2: 33 (invalid resource handle)
flameoffirefly opened this issue · 0 comments
flameoffirefly commented
在deep/feature_extractor.py 里的self.device = "cuda" if torch.cuda.is_available() and use_cuda else "cpu"这句出现该错误,导致后面没有检测结果,能解答下吗?感谢