我已经在我的机器上安装了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错误。
发布于 2019-06-08 23:49:07
您需要将NVIDIA GPU设置为每个操作的默认GPU (在Nvidia控制面板中),或者设置Python应与NVIDIA GPU一起运行(也在Nvidia管理器中)。否则,默认情况下,计算机将自动启动内置英特尔GPU。
https://stackoverflow.com/questions/56507530
复制相似问题