MXNetError: [05:53:50] src/operator/nn/./cudnn/cudnn_convolution-inl.h:287

insightface train.py 报错:mxnet.base.MXNetError: [05:53:50] src/operator/nn/./cudnn/cudnn_convolution-inl.h:287: Check failed: e == CUDNN_STATUS_SUCCESS (8 vs. 0) : cuDNN: CUDNN_STATUS_EXECUTION_FAILED

解决方案:可能是超过显存大小了,改小bach_size=36可以正常运行,本人用的显卡是RTX2080TI

posted @ 2019-09-19 14:05  刘文华  阅读(906)  评论(0编辑  收藏  举报