python - TensorFlow 无法使用 gpu

Question

我刚开始使用 TensorFlow，但我无法在 TensorFlow 1.2.1 中使用 GPU 而不是 CPU。

我有一台配备 NVIDIA GTX 850M 的笔记本电脑，它与 CUDA 5.0 兼容。

CUDA 工具包安装了可用的最新版本。

cuDNN 安装了可用的最新版本。

我已经设置了环境变量，如下所示:https://nitishmutha.github.io/tensorflow/2017/01/22/TensorFlow-with-gpu-for-windows.html

如果我通过 pip 安装最新版本的 TensorFlow:在 cmd 提示符下输入“pip install tensorflow-gpu”，那么 TensorFlow 无法识别我的 GPU，就像我没有:'Device mapping: no known device' .

如果我改为通过“pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl”安装tensorflow ' 然后一切正常。

有谁知道为什么最新版本的 TF 会那样做？

Answer 1

在最新版本的 Tensorflow 中，您可以检查 GPU 可用性

gpu_available = tf.test.is_gpu_available()
is_cuda_gpu_available = tf.test.is_gpu_available(cuda_only=True)
is_cuda_gpu_min_3 = tf.test.is_gpu_available(True, (3,0))

tf.test.is_gpu_available 将在未来版本中删除。更新说明:改用tf.config.list_physical_devices('GPU')