为什么我的tensorflow-gpu在Intel-HD GPU而不是NVIDIA上运行？

Question

我在我的机器上安装了tensorflow-gpu。 在检查设备列表中是否显示GPU时：

from tensorflow.python.client import device_lib
  print(device_lib.list_local_devices())

它清楚地表明NVIDIA 1050 Ti GPU在列表中。 但是当训练数据集和检查任务管理器时，我得到了这个

这怎么可能呢？ 为什么tensorflow使用INTEL-HD而不是NVIDIA？ 它还显示训练时的OOM错误。

Answer 1

您需要将NVIDIA GPU设置为每个操作的默认GPU（在Nvidia控制面板中）或设置Python应该使用NVIDIA GPU（也在Nvidia管理器中）运行。 否则，计算机将默认自动启动内置Intel GPU。