python - mac - install tensorflow with gpu

Cómo ejecutar Tensorflow en la CPU

Question

He instalado la versión de GPU de tensorflow en Ubuntu 14.04.

Estoy en un servidor de GPU donde tensorflow puede acceder a las GPU disponibles.

Quiero ejecutar tensorflow en las CPU.

Normalmente puedo usar env CUDA_VISIBLE_DEVICES=0 para ejecutar en la GPU no. 0.

¿Cómo puedo elegir entre las CPU en su lugar?

No estoy interesado en reescribir mi código with tf.device("/cpu:0"):

Answer 1

Para mí, solo configurar CUDA_VISIBLE_DEVICES con precisión -1 funciona:

Trabajos:

import os
import tensorflow as tf
os.environ[''CUDA_VISIBLE_DEVICES''] = ''-1''
if tf.test.gpu_device_name():
print(''GPU found'')
else:
print("No GPU found")
# No GPU found

No funciona:

import os
import tensorflow as tf
os.environ[''CUDA_VISIBLE_DEVICES''] = ''''
if tf.test.gpu_device_name():
print(''GPU found'')
else:
print("No GPU found")
# GPU found

Answer 2

Puede aplicar el parámetro tf.Session por tf.Session . tf.Session :

config = tf.ConfigProto(
device_count = {''GPU'': 0}
)
sess = tf.Session(config=config)

Ver también archivo de configuración protobuf:

tensorflow/core/framework/config.proto

Answer 3

Si las respuestas anteriores no funcionan, intente:

os.environ[''CUDA_VISIBLE_DEVICES''] = ''-1''

Answer 4

Simplemente usando el código a continuación.

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ[''CUDA_VISIBLE_DEVICES''] = ''-1''

Answer 5

También puede establecer la variable de entorno en

CUDA_VISIBLE_DEVICES=""

sin tener que modificar el código fuente.