首页 > 解决方案 > 如何强制TensorFlow同时使用同一个GPU的所有计算部分?

问题描述

我有一个 GPU 有两个计算部分,但其中只有一个是计算(计算 0),第二个部分不工作(计算 1)。如何强制 TensorFlow 同时使用这两个部分进行计算?我需要完全使用我的 GPU 的容量。请注意,这个问题不是关于两个 GPU,而是关于使用一个 GPU 的两个部分,如下图所示。下图是一个 GPU,但 TensorFlow 正在使用其中的一半,第二部分尚未使用。

在此处输入图像描述

标签: performancetensorflowparallel-processinggpu

解决方案


推荐阅读