pytorch显存一直变大

在代码中添加以下两行可以解决:

torch.backends.cudnn.enabled = True

torch.backends.cudnn.benchmark = True

你可能感兴趣的:(pytorch显存一直变大)