intra_op_parallelism_threads参数设置

1 问题

如代码:

  # Create TF session
  if num_threads:
    config_args = dict(intra_op_parallelism_threads=5)  # 1 => 5
  else:
    config_args = {}
  sess = tf.Session(config=tf.ConfigProto(**config_args))

2 解决

https://stackoverflow.com/questions/41233635/meaning-of-inter-op-parallelism-threads-and-intra-op-parallelism-threads 中给出了一些回答。

TensorFlow会根据 intra_op_parallelism_threads 参数的设置,将待实现的任务分配给各个线程。

因此这个参数会控制某项操作加速的最大并行数。

比如,你在TensorFlow中构建了一个计算图。其中有很多独立的计算,它们之间互不关联。
那么TensorFlow会根据 intra_op_parallelism_threads 设置的线程数,来同时运行它们。

参数的默认值是0,此时系统会选择合适的线程数,也就是会从每一个CPU核心中都选择一个线程来执行任务。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值