Python TensorRT:CUDNN_STATUS_映射_ERROR E

2024-10-01 09:28:01 发布

您现在位置:Python中文网/ 问答频道 /正文

当使用TensorRT的Python API(以及PyCUDA)运行面部识别算法时,我遇到以下错误:

[TensorRT] ERROR: ../rtSafe/safeContext.cpp (133) - Cudnn Error in configure: 7 (CUDNN_STATUS_MAPPING_ERROR)

[TensorRT] ERROR: FAILED_EXECUTION: std::exception

代码仍在编译和运行,但结果不准确——当期望更连续的数字范围时,程序的输出在0.99999907141816之间波动。我已经用TF-TRT和Keras进行了测试,我的代码在这两种情况下都可以工作(只做了一些小的修改,以适应TF和kerasapi之间的差异)。在

我试过安装不同版本的CUDA(9.0、10.0和10.1)和CuDNN(7.6.3、7.6.5)。TensorRT版本是6.0.1.5,PyCUDA是2019.1.2。如果有帮助的话,我会在ubuntu18.04上运行这个。在

任何帮助都将不胜感激!在

更新:我认为错误是由同时运行TensorFlow会话引起的。具体来说,我使用的mtcnn包(link)可能会干扰TensorRT。当mtcnn初始化TF会话时,会发生上述错误;当mtcnn未使用时,不会发生此错误,并且一切都按预期运行。在


Tags: 代码pycuda版本算法apitf错误error