python - Tensorflow 不使用 GPU

Question

我已经使用 pip (在 Anaconda 内)安装了 tensorflow-gpu。验证如下:

$ 点列表 | grep tensorflow tensorflow GPU (1.4.0)

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

我尝试过的另一个命令:

$ 查找 . -名称 tensorflow ./anaconda3/lib/python3.6/site-packages/tensorflow ./anaconda3/lib/python3.6/site-packages/tensorflow/include/tensorflow

有冲突吗？

Answer 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