为啥我的 tensorflow-gpu 在 Intel-HD GPU 而不是 NVIDIA 上运行?
Posted
技术标签:
【中文标题】为啥我的 tensorflow-gpu 在 Intel-HD GPU 而不是 NVIDIA 上运行?【英文标题】:Why is my tensorflow-gpu running in Intel-HD GPU instead of NVIDIA?为什么我的 tensorflow-gpu 在 Intel-HD GPU 而不是 NVIDIA 上运行? 【发布时间】: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 而不是主板。 【参考方案1】:您需要将 NVIDIA GPU 设置为每个操作的默认 GPU(在 Nvidia 控制面板中)或设置 Python 应与 NVIDIA GPU 一起运行(也在 Nvidia 管理器中)。否则电脑会默认自动启动内置Intel GPU。
【讨论】:
以上是关于为啥我的 tensorflow-gpu 在 Intel-HD GPU 而不是 NVIDIA 上运行?的主要内容,如果未能解决你的问题,请参考以下文章
tensorflow-gpu 不适用于 Blas GEMM 启动失败