TensorFlow：对cuInit的调用失败：CUDA_ERROR_NO_DEVICE

Question

我的测试：

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

错误：

C：\升\工作\ tensorflow-1.1.0 \ tensorflow \ stream_executor \ CUDA \ cuda_driver.cc：405]

调用cuInit失败：CUDA_ERROR_NO_DEVICE

- ＆GT;但“/ cpu：0”工作正常

配置：

nvidia-smi：

• CUDA版本9.1
• tensorflow-1.1.0
• Windows 10
• cudnn64_7.dll（安装在C：\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v9.1 \ bin）
• 我的Conda环境中只安装了tensorflow-gpu

为什么Tensorflow无法检测到我的GPU？

Answer 1

问题已解决on GitHub。如果您为CUDA_VISIBLE_DEVICES环境变量设置了无效值，则会显示此错误消息，例如当您只有一个GPU（标识为0）并设置CUDA_VISIBLE_DEVICES=1或CUDA_VISIBLE_DEVICES=2时。

Answer 2

真正的答案是添加以下代码行：

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"