在主内存和GPU内存上分配TensorFlow操作?

时间:2016-08-26 09:23:32

标签: memory-management tensorflow gpu cpu

在购买大型显卡之前,我想在较小的机器上测试网络 检查其性能和对我试图解决的计算机视觉问题的适用性。

我的笔记本电脑配有GTX950M(2GB内存)和8GB主内存。我估计网络的内存消耗大约为2GB(考虑到每个操作和变量),因此我的想法是将大部分操作和变量外包到主内存:

with tf.device('/cpu:0'):
  some_operations()

但网络无法为渐变分配内存。它还依赖于仅为gpu实现的操作,因此仅使用cpu将无法工作......

with tf.device('/cpu:0'): (...)实际上是使用主内存还是仅将计算与cpu相关联?

有没有办法达到我想要的目的?

0 个答案:

没有答案