Warning: file_get_contents(/data/phpspider/zhask/data//catemap/2/tensorflow/5.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
Memory management 在主内存和GPU内存上分配TensorFlow操作？_Memory Management_Tensorflow_Gpu_Cpu - Fatal编程技术网
Fatal编程技术网
  • memory-management/
  • Memory management 在主内存和GPU内存上分配TensorFlow操作？

Memory management 在主内存和GPU内存上分配TensorFlow操作？

memory-management tensorflow

Memory management 在主内存和GPU内存上分配TensorFlow操作？,memory-management,tensorflow,gpu,cpu,Memory Management,Tensorflow,Gpu,Cpu,在购买大型图形卡之前，我想在较小的机器上测试网络 为了检查它的性能和在我试图解决的计算机视觉问题上的适用性 我有一个带有GTX950M（2GB内存）和8GB主内存的笔记本。我估计网络的内存消耗约为2GB（考虑到每个操作和变量），因此我们的想法是使用以下方法将大部分操作和变量外包给主内存： with tf.device('/cpu:0'): some_operations() 不过，网络无法为渐变分配内存。此外，它还依赖于仅为gpu实现的操作，因此仅使用cpu将无法工作 使用tf.devic

在购买大型图形卡之前，我想在较小的机器上测试网络 为了检查它的性能和在我试图解决的计算机视觉问题上的适用性

我有一个带有GTX950M（2GB内存）和8GB主内存的笔记本。我估计网络的内存消耗约为2GB（考虑到每个操作和变量），因此我们的想法是使用以下方法将大部分操作和变量外包给主内存：

with tf.device('/cpu:0'):
  some_operations()
不过，网络无法为渐变分配内存。此外，它还依赖于仅为gpu实现的操作，因此仅使用cpu将无法工作

使用tf.device（'/cpu:0'）：（…）的
是实际使用主内存还是仅将计算与cpu关联


有什么方法可以实现我想要的吗？
op也会分配内存，所以如果op固定在
cpu:0
上，它将使用main memoryThanks@YaroslavBulatov。我会再看一次我的记忆评估，也许我只是把事情搞砸了。我还用
tf.device（'/cpu:0'）
创建了TensorFlow变量，这也应该把它们放在主内存中，对吗？（我使用
tf.get_变量（…）
）op也分配内存，因此如果op被固定到
cpu:0
，它将使用main memoryThanks@YaroslavBulatov。我会再看一次我的记忆评估，也许我只是把事情搞砸了。我还用
tf.device（'/cpu:0'）
创建了TensorFlow变量，这也应该把它们放在主内存中，对吗？（我使用
tf.get_变量（…）
）