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