windows - tensorflow : failed call to cuInit: CUDA_ERROR_NO_DEVICE

Question

我的测试:

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

错误:

c:\l\work\tensorflow-1.1.0\tensorflow\stream_executor\cuda\cuda_driver.cc:405]

调用 cuInit 失败:CUDA_ERROR_NO_DEVICE

-> 但是“/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(ID 0)并设置 CUDA_VISIBLE_DEVICES=1 或 CUDA_VISIBLE_DEVICES=2 时。