张量循环运算：内存溢出原因及解决

写在前面：本博客仅作记录学习之用，部分图片来自网络，如需引用请注明出处，同时如有侵犯您的权益，请联系删除！

---

---

内存溢出

使用AlexNet遍历大量图像进行指标运算（LPIPS），内存溢出。

解决方法

• 溢出原因：循环多次，没有释放变量，导致一直内存占用增加。
• tensor.item(),减少内存溢出。
  每次循环由10Mb降低到0.6Mb
  
  内存占用由98%降到50%

当然也可能出现在比如说损失loss爆显存，去item()可能也会起作用，但是如果需要进行反向传播的时候，慎用。

---

致谢

欲尽善本文，因所视短浅，怎奈所书皆是瞽言蒭议。行文至此，诚向予助与余者致以谢意。