在CPU上运行Tensorflow

我在Ubuntu 14.04上安装了tensorflow的GPU版本。

我在一个GPU服务器上，tensorflow可以访问可用的GPU。

我想在CPU上运行tensorflow。

通常我可以使用env CUDA_VISABLE_DEVICE=0在GPU编号上运行。 0。

我怎样才能在CPU之间select呢？

我没有with tf.device("/cpu:0"):重写我的代码with tf.device("/cpu:0"):

Answer 1

您可以应用tf.Session device_count参数：

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

另请参阅protobufconfiguration文件：

 tensorflow/core/framework/config.proto 

Answer 2

你也可以设置环境variables

 CUDA_VISIBLE_DEVICES="" 

而不必修改源代码。

Answer 3

如果上述答案不起作用，请尝试：

 os.environ['CUDA_VISIBLE_DEVICES'] = '' 

要么

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