Tensorflow gpu版本耗尽内存，配备16GB RAM和8 GB GPU Nvidia Quadro M4000

Question

我正在使用GPU和Nvidia Quadro M4000 8GB显卡进行张量调整。 我遇到了内存错误问题。我尝试过使用内存增长方法，但仍然面临同样的问题。

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

这里形状[1,64,1228,1029]位于第一个汇集层之后

Windows 7上已经安装了所有内容：

• python3.5.2
• 的virtualenv
• cuda toolkit 8.0
• cudnn 5.1

Answer 1

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

尝试查看您在程序中分配的其他张量。很可能你正在分配许多这样的大小，或者你的程序中的许多其他地方。

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