Comment faire fonctionner Tensorflow sur le CPU

Question

J'ai installé la version GPU de tensorflow sur une Ubuntu 14.04.

Je suis sur un serveur GPU où tensorflow peut accéder aux GPU disponibles.

Je veux exécuter tensorflow sur les CPUs.

Normalement, je peux utiliser env CUDA_VISIBLE_DEVICES=0 pour fonctionner sur le GPU no. 0.

Comment puis-je choisir entre les processeurs à la place ?

Je ne suis pas intéressé par la réécriture de mon code avec with tf.device("/cpu:0"):

Answer 1

Dans certains systèmes, il faut spécifier :

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]=""  # or even "-1"

AVANT d'importer tensorflow.