开发者

基于CUDA out of memory的一种神奇解决方式

开发者 https://www.devze.com 2023-02-23 09:20 出处:网络 作者: JasmineFeng
目录CUDA out of memory的解决CUDA error: out of memory问题解决思路溯寻总结CUDA out of memory的解决
目录
  • CUDA out of memory的解决
  • CUDA error: out of memory问题
    • 解决思路溯寻
  • 总结

    CUDA out of memory的解决

    我的输入样本维度是(1000,12,24,开发者_C入门72),一开始我是这么输入数据的:

    trainer.train(x_train, t_train, x_val, t_val)

    发现必溢出,后来我取出其中400个样本输入:

    trainer.train(x_train[:400], t_train[:400], x_val, t_val)

    发现不溢出了,训练正js常,然后我把400删掉,但没删冒号:

    trainer.train(x_train[:], t_train[:], x_val, t_val)

    竟然也没有溢出!!!虽然训练速度降了,但是也能正常训练,我不是很懂原理是啥,好神奇!

    但是样本量一大起来,即使用冒号法还是可能会溢出。比方说我js后来把4000个样本作为x/t_train:

    >>>x_train.shape
    (4000,12,24,72)
    >>>t_train.shape
    (4000,24)
    >>>trainer.train(x_train[:], t_train[:], x_val, t_val)
    RuntimeError:CUDA out of memory...

    之所以说“可能&rdandroidquo;会溢出,是因为确实是有小概率能正常训练的,但是大部分情况还是会out of memory…

    不过这真的是一个“神奇”的方法,能从一编程客栈定程度上解决CUDA out of memory 的问题。希望大佬能解释一下这其中的原理~谢谢!

    -------------------------------------------------------------------

    再附上两个我昨天看到的两种不同的解决方案。

    一个是减少带梯度的中间变量(即www.devze.com非叶子节点)。简言之,能一行代码搞定的,尽量不要写成多行,即使写成多行,也要尽可能减少新变量的建立。

    另一个是在eval的时候,让所有的变量都不带梯度。只需要添加一行代码:

    with torch.no_grad():
     outputs = Net_(inputs)

    在with语句里的所有变量requires_grad都是False。

    CUDA error: out of memory问题

    本人遇到的问题是在训练是正常,一到验证时就会出现cuda error: out of memory的问题

    解决思路溯寻

    1.首先就是考虑减少BATch_size和num_worker,对于我的情况不奏效

    2.然后找到pin_memory发现是设置的True,改为false,仍旧不管用

    3.包括把

    # Empty GPU cache
        if torch.cuda.is_available():
          torch.cuda.empty_cache()

    放到报错位置的前后,不奏效

    4.后来再聚焦问题关键,是一到验证就会出问题,所以专门查攻略,我初步怀疑是因为验证没有参与反向传播,梯度累积,内存爆了,但当时代码中有with torch.no_grad(): ,所以并没有发现关键,知道看到别人里面 forword是放在with torch.no_grad()后面的,所以最后

    with torch.no_grad():
            # Forward pass
            loss, np_probs, hv_logits = self.forward(images, targets)

    问题解决!

    总结

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持我们。

    0

    精彩评论

    暂无评论...
    验证码 换一张
    取 消

    关注公众号