赞
踩
tensorflow在电脑的gpu和cpu上均可运行:
"/cpu:0"
:机器的 CPU。"/device:GPU:0"
:机器的 GPU(如果有一个)。"/device:GPU:1"
:机器的第二个 GPU(以此类推)。当想要知道指令和张量在哪个设备上运行时,可以这样设置session:
sess = tf.Session(config=tf.ConfigProto(log_device_placemnet=True))
- with tf.device('/cpu:0'):
- ...
- sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
当使用GPU时,需要设置允许GPU增加内存,有两种设置方式:
(1)设置allow_grouth,根据运行时的需要来分配GPU内存
- config = tf.ConfigProto()
- config.gpu_options.allow_growth = True
- session = tf.Session(config=config,...)
(2)设置per_
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。