本文主要是介绍pytorch 显存自适应减少，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
pytorch在训练时，显存是只会增加不会减少的。如果要动态减少：
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
不过这样做的后果就是训练会很慢，快赶上CPU的速度了
这篇关于pytorch 显存自适应减少的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
