TensorFlow GPU内存

问题描述 投票:1回答:1

我有一个非常大的深度神经网络。当我尝试在GPU上运行它时,我会在分配时获得“OOM”。但是当我屏蔽GPU并在CPU上运行时,它可以工作(比较小型号时速度大约慢100倍)。

我的问题是,如果tenosrflow中有任何机制可以让我在GPU上运行模型。我假设CPU使用虚拟内存,因此它可以根据自己的喜好进行分配,并在缓存/ RAM /磁盘(抖动)之间移动。

Tensorflow与GPU有类似之处吗?这对我有帮助,即使它比常规GPU运行慢10倍

谢谢

tensorflow memory-management gpu
1个回答
0
投票

GPU内存目前无法扩展(直到像PASCAL这样的东西可用)

© www.soinside.com 2019 - 2024. All rights reserved.