pytorch显存一直变大

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

torch.backends.cudnn.enabled = True

torch.backends.cudnn.benchmark = True