pytorch 減少顯存消耗，優化顯存使用，避免out of memory

本文是整理了大神的兩篇博客：python 如何計算模型以及中間變量的顯存佔用大小： git https://oldpan.me/archives/how-to-calculate-gpu-memory github 如何在Pytorch中精細化利用顯存： 算法 https://oldpan.me/archives/how-to-use-memory-pytorch網絡 還有知乎中大神的解答：app

>>阅读原文<<