在代码中添加以下两行可以解决：
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
在代码中添加以下两行可以解决：
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
本文标题：pytorch显存一直变大
本文链接：https://www.haomeiwen.com/subject/pcecgqtx.html
网友评论