我有2台计算机使用相同的GPU Quadro M4000,均为8GB。我在两台计算机上运行相同的train.prototxt,一个可以运行批量大小= 5(因为我的图像大小很大),但另一个不能并且给出错误
syncedmem.cpp:51] Check failed: error == cudaSuccess (2 vs. 0) out of memory
有谁知道为什么?
答案 0 :(得分:0)
使用以下命令检查:
nvidia-smi
它将显示在GPUS上运行的进程。比较两台机器的统计数据,您可以检查它在其他机器上运行的原因。可能在一个GPU上有一些消耗内存的其他程序/
答案 1 :(得分:0)
也许其他GPU内存不足。尝试减少批量大小(例如:batch size = 1
)