如何在Pytorch中精细化利用显存

前言 在上篇文章《浅谈深度学习:如何计算模型以及中间变量的显存占用大小》中我们对如何计算各种变量所占显存大小进行了一些探索。而这篇文章我们着重讲解如何利用Pytorch深度学习框架的一些特性,去查看我们当前使用的变量所占用的显存大小,以及一些优化工作。以下代码所使用的平台框架为Pytorch。 优化显存 在Pytorch中优化显存是我们处 … 继续阅读如何在Pytorch中精细化利用显存