Tensorflow看不到GPU设备

时间:2018-05-18 00:29:24

标签: python tensorflow gpu

我不是机器学习和GPU使用的初学者。我有一个很好的工作环境,但最近Windows更新了,并且出现了问题。我有下一个环境:

  • OS:Win10 64
  • Python 3.5.4
  • Tensorflow-gpu 1.7.0
  • Nvidia司机
  • CUDA工具包 - 9.0
  • cuDNN - 7.1 for CUDA 9.0

下一个代码正常工作

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

但是当我尝试使用GPU时,我会给出下一个麻烦:

tensorflow.python.framework.errors_impl.InvalidArgumentError:
Cannot assign a device for operation 'random_uniform/sub':
Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. 
Make sure the device specification refers to a valid device.
[[Node: random_uniform/sub = Sub[T=DT_FLOAT, _device="/device:GPU:0"](random_uniform/max, random_uniform/min)]]

我认为这是因为我没有 Microsoft Visual C ++ 2015 Redistributable Update 3 。但是当我尝试安装它时,我收到消息“已经安装了该产品的另一个版本”。 我试图在谷歌找几个小时的答案,但没有。为什么会出现这个问题?

0 个答案:

没有答案