TensorFlow基础知识点（一）切换CPU或GPU

TensorFlow中文翻译使用手册

以下系列都是或多或少来自手册内容，只是作为易忘点进行整理。

---

如果设备不止一个GPU,需要明确op操作在那个GPU上去调用他们。可以使用with ... Device语句明确指定那个Cpu或Gpu将被调用。

with tf.Session() as ses:
    with tf.device("/gpu:1"):
        matrix1=tf.constant([[3.,3.]])
        matrix2=tf.constant([[2.],[2.]])
        product=tf.matmul(matrix1,matrix2)

---

字符	对应的操作
"/cpu:0"	The CPU of your machine
"/gpu:0"	The GPU of yout machine ,if you have one

等等依次类推：