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：＆＃34; pip install tensorflow-gpu＆＃34;在cmd提示符下，然后TensorFlow无法识别我的GPU并且表现得像我没有：＆＃39;设备映射：没有已知的设备＆＃39;。

如果我通过＆＃39; pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl＆＃39;安装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')

TensorFlow无法使用gpu

1 个答案: