如何设置tensorflow优先使用特定CPU和GPU进行运算？

本文共计616个文字，预计阅读时间需要3分钟。

1. 指示GPU运算：若安装的是GPU版本，TensorFlow在运行过程中能自动检测。检测到GPU后，TensorFlow会尽可能利用找到的第一块GPU执行操作。若机器上有多个可用GPU，TensorFlow将优先使用第一个。

1.指定GPU运算

如果安装的是GPU版本，在运行的过程中TensorFlow能够自动检测。如果检测到GPU，TensorFlow会尽可能的利用找到的第一个GPU来执行操作。

如果机器上有超过一个可用的GPU，除了第一个之外的其他的GPU默认是不参与计算的。为了让TensorFlow使用这些GPU，必须将OP明确指派给他们执行。with......device语句能够用来指派特定的CPU或者GPU执行操作：

import tensorflow as tf import numpy as np with tf.Session() as sess: with tf.device('/cpu:0'): a = tf.placeholder(tf.int32) b = tf.placeholder(tf.int32) add = tf.add(a, b) sum = sess.run(add, feed_dict={a: 3, b: 4}) print(sum)

设备的字符串标识，当前支持的设备包括以下的几种：

cpu:0 机器的第一个cpu。