在我安装第二块显卡之前,我已成功训练了一个使用张量流的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工作顺利。
有什么问题?