关于深度学习神经网络模型训练,参数过大,导致显卡内存溢出问题的总结

news/2024/10/17 22:24:54/

实验是利用CIFAR10模型,计算将图片分成10个分类的误差。一开始采用cpu模型来训练,迭代一个epoch需要50分钟,所以换成GPU模型,GPU有4G内存,结果刚加载模型显卡内存就溢出。

报错如下:

先用keras来计算网络的大小,参数的个数。

(50000, 32, 32, 3) (50000,) (10000, 32, 32, 3) (10000,)
sample: (512, 32, 32, 3) (512,) tf.Tensor(-1.0, shape=(), dtype=float32) tf.Tensor(1.0, shape=(), dtype=float32)
Model: "res_net_1"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
========================================&#

http://www.ppmy.cn/news/714185.html

相关文章

显卡 内存分配 linux,【原创】Linux环境下的图形系统和AMD R600显卡编程(4)——AMD显卡显存管理机制...

显卡使用的内存分为两部分,一部分是显卡自带的显存称为VRAM内存,另外一部分是系统主存称为GTT内存(graphics translation table和后面的GART含义相同,都是指显卡的页表,GTT 内存可以就理解为需要建立GPU页表的显存)。在嵌入式系统或者集成显卡上,显卡通常是不自带显存的,…

win10弹出计算机的内存不足,win10提示计算机显卡内存不足情况的解决办法介绍...

近日,有网友反映,windows10系统在内存充足的情况下,出现提示计算机的内存不足的对话框现象,那么,提示win10计算机内存不足是什么原因?如何解决?针对此问题,本文就为大家进行解答。有…

计算机运行内存和显卡内存不足,电脑内存不足的解决办法 win10提示计算机显卡内存不足情况的解决办法介绍 我这个...

延伸:win10提示计算机显卡内存不足情况的解决办法介绍 我这个配置玩坦克世界提示内存 描述:近日,有网友反映,windows10系统在内存充足的情况下,出现提示计算机的内存不足的对话框现象,那么,提示win10计算机内存不足是什么原因?如何解决?针对此问题,本文就为大家进行解…

显卡内存一直被占用解决方式

lsof /dev/nvidia* kill相应pid http://stackoverflow.com/questions/4354257/stop-all-cuda-processes-in-linux-without-restarting-the-computer

显卡(N卡)内存被占用如何处理?

输入Nvidia-smi 比如这里面PID:20779占了很大的内存,但程序又没有被停掉,我允许其他程序的时候,就会爆内存(out of memory)。 然后 kill - 9 PID 就可以了,很好用。 参考资料: NVIDIA-SMI详解…

ubuntu gpu显卡内存 一直被占,nvidia-smi 看不到进程

使用 fuser 命令 查看 显卡进程信息,注意调整参数,可能你的显卡 是nvidia1,或nvidia6 fuser -v /dev/nvidia0 如果 fuser 命令 找不到,如图 fuser: command not found 安装fuser shell命令 apt-get install psmisc 查找到 …

使用YOLOv5模型训练: 显卡内存不够导致报错RuntimeError: CUDA out of memory

Logging results to runs\train\exp18 Starting training for 300 epochs...Epoch gpu_mem box obj cls labels img_size0%| | 0/8 [00:03<?, ?it/s] Traceback (most recent call last):File "E:\SelfLearning\Pytorch_Learning\y…

显卡内存管理机制及驱动实现(Intel gma500为例)

背景 在Intel Atom芯片的环境中分析故障时&#xff0c;顺便看了下该环境的显卡驱动中的内存管理相关机制&#xff0c;显卡为Atom CPU的集成显卡&#xff0c;低端产品&#xff0c;gma500显卡。本文主要关注该显卡驱动中的内存管理相关的实现原理&#xff0c;其他显卡原理与之类…