关于torch开启cudnn导致显存占用过多的问题

1、问题

基于retinaface+insightface封装了一个人脸识别接口,两个模型一起跑的时候显存占用为2.5G,但是将人脸识别模块删除后,显存占用约为3.5G

2、解决方法

import torch.backends.cudnn as cudnn
cudnn.benchmark = True

将这两句话注释掉,人脸检测+识别模型占用显存约1.5G,

但是注释掉识别功能,显存占用为1.3G左右

你可能感兴趣的:(关于torch开启cudnn导致显存占用过多的问题)