python - instalar - install tensorflow windows

Error de requisito de dispositivo explícito de flujo de Tensor

Question

Estoy tratando de ejecutar el tutorial CIFAR10 con el código de entrenamiento en una GPU y el código de evaluación en la otra. Sé con certeza que tengo dos gpus en mi computadora, y puedo probar esto ejecutando los ejemplos simples aquí: https://www.tensorflow.org/how_tos/using_gpu/index.html

Sin embargo, usar a with device(''/gpu:0'') no funciona para la mayoría de las variables en el ejemplo de CIFAR. Intenté muchas combinaciones de diferentes variables en gpu vs. cpu, o todas las variables en uno u otro. Siempre el mismo error para alguna variable, algo como esto:

Cannot assign a device to node ''shuffle_batch/random_shuffle_queue'': Could not satisfy explicit device specification ''/gpu:0''

¿Es posible que esto sea un error en Tensor Flow o me falta algo?

Answer 1

Could not satisfy explicit device specification lo que significa que no tienes el dispositivo correspondiente. ¿Realmente tiene una GPU habilitada para CUDA en su máquina?

ACTUALIZACIÓN: como se vio en la discusión a continuación, este error también se plantea si la operación particular (en este caso, RandomShuffleQueue ) no se puede ejecutar en la GPU, ya que solo tiene una implementación de CPU.

Si está de acuerdo con que TensorFlow elija un dispositivo para usted (en particular, allow_soft_placement a la CPU cuando no está disponible la implementación de GPU), considere establecer allow_soft_placement en su configuración, según este artículo :

sess = tf.Session(config=tf.ConfigProto(
    allow_soft_placement=True, log_device_placement=True))