CUDA out of memory. (解决了)

最新推荐文章于 2024-08-17 22:11:19 发布

zhaoyuyu_nudt

最新推荐文章于 2024-08-17 22:11:19 发布

阅读量2.7k

点赞数 1

分类专栏：实验日志

本文链接：https://blog.csdn.net/yinxian9019/article/details/108269973

版权

本文介绍了遇到CUDA out of memory错误的原因，包括系统未释放内存和单个样本占用过多内存。提出了多GPU训练时的问题，如负载不均衡，并给出了解决办法，包括删除不必要的变量、减小batch_size、优化pytorch版本、平衡GPU负载以及调整输入尺寸等策略。

摘要由CSDN通过智能技术生成

报错

Traceback (most recent call last):
  File "plastic_train.py", line 243, in <module>
    y, hebb = plasticnet.module(Variable(inputs[numstep], requires_grad=False), y, hebb)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 541, in __call__
    result = self.forward(*input, **kwargs)
  File "/mnt/home/Track/PlasticNet/UpdateNet/updatenet/net_pld.py", line 75, in forward
    hebb = (1 - self.eta) * hebb + self