当使用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
未使用时,不会发生此错误,并且一切都按预期运行。在
目前没有回答
相关问题 更多 >
编程相关推荐