tf.device

如果需要切换成CPU运算，可以调用tf.device(device_name)函数，其中device_name格式如/cpu:0其中的0表示设备号，TF不区分CPU的设备号，设置为0即可。GPU区分设备号\gpu:0和\gpu:1表示两张不同的显卡。 
在一些情况下，我们即使是在GPU下跑模型，也会将部分Tensor储存在内存里，因为这个Tensor可能太大了，显存不够放，相比于显存，内存一般大多了，于是这个时候就常常人为指定为CPU设备。这种形式我们在一些代码中能见到。如：

with tf.device('/cpu:0'):
    build_CNN() # 此时，这个CNN的Tensor是储存在内存里的，而非显存里。

需要注意的是，这个方法会减少显存的负担，但是从内存把数据传输到显存中是非常慢的，这样做常常会减慢速度。

原文：https://blog.csdn.net/LoseInVain/article/details/78814091 
 

tf.device()指定tensorflow运行的GPU或CPU设备版本：

https://blog.csdn.net/dcrmg/article/details/79747882

基础知识

https://blog.csdn.net/lemon_tree1994/article/details/79206790