用GPU做caffe训练时提示:out of memory,查看了一下显存占用情况好像不太够,显卡是1070,显存8G

用GPU做caffe训练时提示:out of memory,查看了一下显存占用情况好像不太够,显卡是1070,显存8G,是否有不更换显卡来减小显存占用的方法,不知道安装cudnn能否减小显存占用?
已邀请:

Evence

赞同来自: 不卑不亢ZJX

把你的batch_size设小一点试试,你可以先尝试为1,显存够了之后再试4、8、16、32、64……
具体在你的train.prototxt中的Data层

batchsize.jpg

 

啊边边

赞同来自:

打扰了楼主,我的也遇到了同样的问题,请问您后来是怎么解决的?

陶潜水

赞同来自:

可以减少batchsize,也可以换一个优化的Caffe版本进行训练,比如nvidia优化过的caffe

要回复问题请先登录注册