咖啡|检查失败:错误== cudaSuccess(2 vs. 0)内存不足
caffe
deep-learning
neural-network
5
0

我正在尝试在Caffe上训练网络。我的图片尺寸为512x640。批处理大小为1。我正在尝试实现FCN-8

我目前正在具有4GB GPU内存的Amazon EC2实例(g2.2xlarge)上运行此程序。但是当我运行求解器时,它立即抛出错误

Check failed: error == cudaSuccess (2 vs. 0)  out of memory
*** Check failure stack trace: ***
Aborted (core dumped)

有人可以帮我从这里继续吗?

参考资料:
Stack Overflow
收藏
评论
共 2 个回答
高赞 时间 活跃

Caffe可以使用多个GPU。仅C ++接口支持此功能,而python接口不支持。您也可以启用cuDNN以减少内存占用。

https://github.com/BVLC/caffe/blob/master/docs/multigpu.md

收藏
评论

您得到的错误确实是内存不足,但不是RAM,而是GPU内存(请注意,该错误来自CUDA)。
通常,当caffe内存不足时-首先要做的是减小批次大小(以渐变精度为代价),但是由于您已经是批次大小= 1 ...
你确定批量大小为1 两种训练和测试阶段?

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号