TensorFlow:对cuInit的调用失败:CUDA_ERROR_NO_DEVICE

时间:2018-02-07 07:31:00

标签: windows tensorflow cudnn

我的测试:

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

- >但“/ cpu:0”工作正常

配置:

nvidia-smi: enter image description here

  • 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?

2 个答案:

答案 0 :(得分:8)

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

答案 1 :(得分:5)

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

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