TensorFlow 不使用 GPU

Posted 2023-04-15

【中文标题】TensorFlow 不使用 GPU

【英文标题】：Tensorflow not using the GPU

【发布时间】：2018-07-13 14:33:36

【问题描述】：

我已经使用pip（在 Anaconda 中）安装了tensorflow-gpu。这是验证：

$ pip list | grep tensorflow tensorflow-gpu (1.4.0)

但每当我运行模型时，它都会使用 CPU。有人可以帮忙吗。

我尝试的另一个命令：

$ find . -name tensorflow ./anaconda3/lib/python3.6/site-packages/tensorflow ./anaconda3/lib/python3.6/site-packages/tensorflow/include/tensorflow

有冲突吗？

【问题讨论】：

为什么帖子旁边有一个“-1”？

Tensorflow not running on GPU的可能重复

遇到了类似的问题。我的情况是使用 conda 安装了 tensorflow-gpu，而 pip 安装了 tensorflow cpu 版本。我使用 pip 重新安装了 tensorflow-gpu 解决了这个问题。

【参考方案1】：

假设您使用的是 Nvidia-gpu，您在安装支持 gpu 的 Tensorflow 之前是否安装了 cuda 和 cudnn？检查这个link。另外使用 pip 或 pip3 安装 tensorflow，因为 Anaconda 不会有最新版本的 tensorflow。

还可以尝试在 python 或 ipython shell 中运行以下命令

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

您将获得一个示例输出，如果您在消息中看到您的 GPU 设备，则说明 tensorflow 正在使用您的 gpu

【讨论】：

我也试过了，我可以在列表中看到我的 gpu。我有一个微星。使用我的 Dragon 中心，它显示了我的硬件功能的统计数据和指标，我可以看到我的 GPU 没有被使用，而是我所有的 RAM 都被使用了。是不是指标无法选择的东西？此外，当我运行我的游戏时，我可以看到 GPU 正在被使用。

@LakshmiNarayanan 你有 nvidia GPU 还是 AMD？