问TensorFlow图形处理器内存
EN

Stack Overflow用户

提问于 2018-06-16 00:40:11

回答 1查看 125关注 0票数 1

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

我的问题是，tenosrflow中是否有任何机制可以让我在GPU上运行模型。我假设CPU使用虚拟内存，所以它可以随心所欲地分配，并在缓存/RAM/磁盘之间移动(颠簸)。

在Tensorflow和GPU上有相似的东西吗？即使它比常规GPU运行速度慢10倍，这也会对我有所帮助

谢谢

发布于 2019-03-14 00:38:00

图形处理器内存目前不可扩展(直到PASCAL之类的东西可用)

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/50879688

复制

相似问题

问TensorFlow图形处理器内存EN