擅长:python、mysql、java
<p>这就是张量流的工作原理。当它开始使用GPU时,它几乎分配了所有的内存。你知道吗</p>
<p>您可以尝试的一件小事是限制GPU分配的部分:</p>
<pre><code>gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction = 0.5)
sess = tf.Session(config = tf.ConfigProto(gpu_options = gpu_options))
</code></pre>
<p>但它控制所有GPU内存,因此您无法确定内存将如何分割(如果将0.25设置为1 GPU上的所有内存,将0设置为其他GPU或其他配置上的所有内存)。你知道吗</p>