相关疑难解决方法(0)

Caffe | 检查失败:错误== cudaSuccess(2对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)
Run Code Online (Sandbox Code Playgroud)

有人可以帮助我从这里开始吗?

amazon-web-services neural-network deep-learning caffe

10
推荐指数
1
解决办法
2万
查看次数