tensorflow:在单GPU计算机中运行良好但在双GPU计算机中运行不良

时间:2017-11-15 18:26:05

标签: python tensorflow multi-gpu

在我安装第二块显卡之前,我已成功训练了一个使用张量流的VGG-11架构模型和一个具有6GB内存的单个GPU。但是当我安装第二块显卡并运行相同的代码(allow_growth = True且没有使用tf.device())时,我收到了OOM错误。

我的理解是我的第二张卡(GPU:0)有8GB内存,当我没有使用tf.device()来指定任何设备时,TF会使用我的“device:gpu:0”默认进行计算。并且内存应该足够了,因为8GB> 6GB。

然后我尝试使用CUDA_VISIBLE_DEVICES = 0来阻止一个并运行相同的代码。 TF工作顺利。

有什么问题?

0 个答案:

没有答案