DeepFaceLab错误,921版out of memory的解决方法。

OOM是DeepFaceLab非常常见的错误,一般发生在训练环节,起因是显存不足(不是内存)。但是有部分群友发现,自己明明有8G,11G的显存,但是依旧提示out of memory。奇怪的是,错误一直出现,训练可以继续。关于这个问题,因为本人的卡是10系列,无法验证,所以一直没有找到解决方法,近期有群友又问到这样的问题,我终于确定了解决方案,给大家分享一下。

 

 

问题描述:

首先,这个错误主要发生在GeForce RTX 2060 Super, 2080Super 等20系列的显卡。是否跟Super有关,我没有确认过。

其次,620以及之前版本没有问题,827,907,921等版本似乎都存在这个问题。

错误出现在训练环节,训练启动后出现如上图的提示failed to …. CUDA_ERROR_OUT_OF_MEMORY:out of memory.  每刷新一条loss记录,就出现一次。

 

解决方法

其实,如果你是一个比较关注DeepFaceLab的github项目的人,应该很早就看到过如下的一个信息。

这条issue其实已经提出了解决方案。 图片中大概说的是,有一个人用的是RTX2080 Super 然后遇到了out of memory的问题。 然后,下面有人提供了一个解决方案。方案就是修改nnlib.py文件中144行,将True改为False

nnlib.py 具体路径为:…\_internal\DeepFaceLab\nnlib\nnlib.py

上图为群友的实况截图^_^~  记住修改源代码不推荐用系统自带的记事本,用Notepad++等工具会比较好。

修改完成后,保存,再次开始训练。此时,已经不再异常提示,整个界面干净舒服了很多。

 

   


******************小尾巴*********************

申明:图文均由deepfakes 中文网原创,转发请注明出处,谢谢!本站仅分享技术,切勿用于非法用途
微信公众号:托尼不是塔克(TonyNoStark),学习各种AI黑科技!

  软件下载:点击进入
  入门教程:点击进入
  深入学习:知识星球
  专业服务:脸魔 F M  

*********************************************

22 Comments

回复 花花 取消回复

您的电子邮箱地址不会被公开。 必填项已用*标注

解决 : *
24 ⁄ 3 =