☰

如何检查我是否正确安装了支持 GPU 的 TensorFlow？

tag-icon • tag-icon gpu nvidia-geforce
如何检查我是否正确安装了支持 GPU 的 TensorFlow？

我安装了支持 GPU 的 Tensorflow，想检查一下我是否真的安装正确。
我尝试了Tensorflow其中写道：

$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42

它产生了相同的结果。但是我在安装过程中遇到了很多错误，并以某种方式解决了它们。这真的足以确保我可以继续使用 Tensorflow 吗？谢谢

答案1

要检查您是否在 TensorFlow 中使用 GPU，请在控制台上运行以下命令python：

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

它会转储你的 GPU 的详细描述

答案2

以下是 NVIDIA 对使用 NVIDIA GPU 和 TensorFlow 的要求：

支持 GPU 的 TensorFlow 版本具有以下要求：

64 位 Linux

Python 2.7

NVIDIA CUDA® 7.5（Pascal GPU 需要 CUDA 8.0）

NVIDIA cuDNN v4.0（最低）或 v5.1（推荐）

您还需要支持计算机功能 3.0 或更高版本的 NVIDIA GPU。

更多信息请见：www.nvidia.com/object/gpu-accelerated-applications-tensorflow-installation.html#sthash.eaN8w5me.dpuf

..从http://www.nvidia.com/object/gpu-accelerated-applications-tensorflow-installation.html

在之前的问题中，您说过您的系统有 NVIDIA GeForce GTX 980M：需要解决屏幕闪烁的问题

制造商规格显示“CUDA：是”（来自 geforce.com/hardware/notebook-gpus/geforce-gtx-980m/specifications）。

“是”不是一个数字。制作卡片时可能只有一个数字，即“1”。

要确定 GPU 选项是否与您的系统兼容，唯一的方法就是尝试。然而，结果并不如您所愿。CPU 可能是您唯一稳定的选择。

最好的祝愿

相关内容