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

时间:2017-05-10 12:40:15

标签: tensorflow windows-7-x64

我正在使用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

1 个答案:

答案 0 :(得分:0)

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

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

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