【发布时间】:2019-10-23 17:46:51
【问题描述】:
我已经在我的机器上安装了 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 错误。
【问题讨论】:
-
您误解了数据,没有任何迹象表明 tensorflow 正在英特尔 GPU 中运行,因为它根本不支持它。
-
转到计算机背面,将显示器电缆插入 GPU 而不是主板。
标签: python tensorflow keras nvidia