Caffe在相同的gpu上运行相同的程序，允许不同的批处理大小

Question

我有两台电脑使用相同的GPU Quadro M4000，都是8GB。我坐同一趟火车。在两台计算机上的Prototxt，一台可以运行批处理大小=5(因为我的图像大小很大)，但另一台不能并给出错误

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

有人知道为什么吗?

Answer 1

检查以下命令:

nvidia-smi

它将显示gpu上运行的进程。比较两台机器的统计数据，您可以检查为什么它不能在其他机器上工作。可能在一个GPU上有一些其他程序消耗内存/

Answer 2

可能另一个GPU内存不足。尝试减少批大小(例如:batch size = 1)