- 在tensorflow中,模型可以在本地的GPU或者CPU上运行,用户可以指定模型运行的设备。
- 通常,如果Tensorflow版本是GPU版本,而且你的电脑上配置有符合条件的显卡,那么在不做任何配置的情况下,模型是默认运行在显卡上的。
- 如果要切换成CPU运行,可以调用tf.device(device_name)函数,其中device_name格式如/cpu:0,其中的0表示设备号,TF不区分CPU,设置为0即可。GPU区分设备号\gpu:0和\gpu:1表示两张不同的显卡。
- 在某些情况下,由于显存有限,即使在GPU下跑模型,也会将Tensor存储在内存里,相对来说,内存比显存大很多,于是就需要人为的设置指定CPU设备。这种形势下,可以看到,如:
with tf.device("/cpu:0"): build_CNN()
需要注意的是,这个方法会减少显存的负担,但从内存把数据传到显存非常慢,会延长耗时。
参考自https://blog.csdn.net/LoseInVain/article/details/78814091