代码之家 › 专栏 › 技术社区 › Acy

Tensorflow训练崩溃超过系统内存的10%,尽管批处理大小为1

gpu tensorflow python

0

Acy · 技术社区 · 7 年前

批量大小为128/64/32的训练,用于在几个阶段之后清空GPU内存。然而,运行随机批训练实际上使程序停留在第一个历元的0%。

    --------------- Epoch 1 ---------------
  0%|                                                                                          | 0/486 [00:00<?, ?it/s]2019-06-18 18:04:58.581233: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:04:59.208729: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:04:59.827425: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:05:00.497830: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:05:01.173273: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.

0 回复 | 直到 7 年前

1

0

Rishabh Sahrawat 7 年前

这个警告是在我的箱子里装缓冲液时发出的。这可能是因为数据量,你使用的神经元数量等等。另一个原因可能是你的CPU RAM内存。检查您有多少内存,以及程序在执行时使用了多少内存。如果内存不足,请尝试增加CPU RAM内存。更多信息,也许这个 issue