Tensorflow gpu 版本 运行 内存不足，16GB RAM 和 8GB GPU Nvidia Quadro M4000

Question

我正在使用 GPU 和 Nvidia Quadro M4000 8GB 显卡开发 tensorflow。 我遇到内存错误问题。我尝试过使用内存增长方法，但仍然面临同样的问题。

ResourceExhaustedError: OOM when allocating tensor with shape[1,64,1228,1029] 

这里的形状 [1,64,1228,1029] 出现在第一个池化层之后

一切都已安装在 Windows 7:

• python3.5.2
• 虚拟环境
• cuda 工具包 8.0
• cudnn 5.1

Answer 1

好吧，因为这只有 323 兆字节，所以分配这个张量不应该是唯一的问题。

尝试查看您在程序中分配的其他张量。您很可能在程序中的某处分配了很多这样的大小，或许多其他大小。

您可能遇到的另一个问题是您同时 运行 多个程序。尝试 运行 nvidia-smi 等同于 Windows 以查看您的卡目前使用了多少内存。