GPU/CPU 选择

有时候我们需要测试单cpu和GPU使用上效率的差别， 如何选择使用CPU还是GPU呢？

 

self.graph = tf.Graph()
self.graph.as_default()
if use_gpu:
    session_conf = tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)
    session_conf.gpu_options.allow_growth = True
else:
    cpu_num = 1   # 根据虚拟机实际情况调整
    session_conf = tf.ConfigProto(device_count={"CPU": cpu_num},
                                          inter_op_parallelism_threads=cpu_num,
                                          intra_op_parallelism_threads=cpu_num,
                                          log_device_placement=False)
self.session = tf.Session(config=session_conf)

需要设置use_gpu 参数， bool型， true为使用gpu， false为使用cpu