tensorflow函数_tf.ConfigProto()

tf.ConfigProto()
主要的作用是配置tf.Session的运算方式,比如gpu运算或者cpu运算.
session_config = tf.ConfigProto(
log_device_placement=True,
inter_op_parallelism_threads=0,
intra_op_parallelism_threads=0,
allow_soft_placement=True)
sess = tf.Session(config=session_config)
log_device_placement=True: 设置为True时,会打印出TensorFlow使用了那种操作
inter_op_parallelism_threads=0: 设置线程一个操作内部并行运算的线程数,比如矩阵乘法,如果设置为0,则表示以最优的线程数处理
intra_op_parallelism_threads=0:设置多个操作并行运算的线程数,比如 c = a + b,d = e + f . 可以并行运算
allow_soft_placement=True:有时候,不同的设备,它的cpu和gpu是不同的,如果将这个选项设置成True,那么当运行设备不满足要求时,会自动分配GPU或者CPU。
使用GPU时,自动达到最大内存:config.gpu_options.allow_growth = True

你可能感兴趣的:(tensorflow)