TensorFlow GPU内存
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了TensorFlow GPU内存相关的知识，希望对你有一定的参考价值。
我有一个非常大的深度神经网络。当我尝试在GPU上运行它时，我会在分配时获得“OOM”。但是当我屏蔽GPU并在CPU上运行时，它可以工作（比较小型号时速度大约慢100倍）。
我的问题是，如果tenosrflow中有任何机制可以让我在GPU上运行模型。我假设CPU使用虚拟内存，因此它可以根据自己的喜好进行分配，并在缓存/ RAM /磁盘（抖动）之间移动。
Tensorflow与GPU有类似之处吗？这对我有帮助，即使它比常规GPU运行慢10倍
谢谢
答案
GPU内存目前无法扩展（直到像PASCAL这样的东西可用）
以上是关于TensorFlow GPU内存的主要内容，如果未能解决你的问题，请参考以下文章
Tensorflow 耗尽 GPU 内存：分配器 (GPU_0_bfc) 尝试分配内存不足
为啥 tensorflow 模块会占用所有 GPU 内存？ [复制]