Caffe在相同的GPU上运行相同的程序允许不同的批量大小

时间:2016-10-28 18:46:23

标签: gpu caffe

我有2台计算机使用相同的GPU Quadro M4000,均为8GB。我在两台计算机上运行相同的train.prototxt,一个可以运行批量大小= 5(因为我的图像大小很大),但另一个不能并且给出错误

syncedmem.cpp:51] Check failed: error == cudaSuccess (2 vs. 0)  out of memory

有谁知道为什么?

2 个答案:

答案 0 :(得分:0)

使用以下命令检查:

nvidia-smi

它将显示在GPUS上运行的进程。比较两台机器的统计数据,您可以检查它在其他机器上运行的原因。可能在一个GPU上有一些消耗内存的其他程序/

答案 1 :(得分:0)

也许其他GPU内存不足。尝试减少批量大小(例如:batch size = 1