TensorFlow: llamada fallida a cuInit: CUDA_ERROR_NO_DEVICE

Mi prueba:

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

Error:

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

llamada fallida a cuInit: CUDA_ERROR_NO_DEVICE

-> pero "/cpu:0" funciona bien

Configuración:

nvidia-smi:

• CUDA versión 9.1


• tensorflow-1.1.0


• ventanas 10


• cudnn64_7.dll (instalado en C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin)


• Solo tensorflow-gpu está instalado en mi entorno Conda

¿Por qué Tensorflow no puede detectar mi GPU?

Solución del problema

El problema se resolvió en GitHub. Este mensaje de error se mostrará si establece un valor no válido para la CUDA_VISIBLE_DEVICESvariable de entorno, por ejemplo, cuando solo tiene una sola GPU (que tiene ID 0) y establece CUDA_VISIBLE_DEVICES=1o CUDA_VISIBLE_DEVICES=2.