Make sure the device specification refers to a valid device

开始调用

with tf.Session() as sess,tf.device('/gpu:0'):

InvalidArgumentError (see above for traceback): Cannot assign a device for operation 'init': Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/cpu:0 ]. Make sure the device specification refers to a valid device.

解决方法：pip uninstall tensorflow

pip install tensorflow-gpu

下载cudnn v6，把include lib bin三个目录拷贝到下面目录：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\cuda

后来就能用了

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/78825001