我如何使用tensorflow使用100%的CPU?

时间:2019-07-03 11:48:31

标签: python multithreading tensorflow python-multithreading

我正在设置脚本。现在,当我运行脚本时,我的CPU使用率是25(占其全部容量的1/4)。

我尝试设置以下变量:

import tensorflow as tf

config = tf.ConfigProto(intra_op_parallelism_threads=4, inter_op_parallelism_threads=2, allow_soft_placement=True, device_count = {'CPU': 4 })

session = tf.Session(config=config)

K.set_session(session)

os.environ["OMP_NUM_THREADS"] = "4"

os.environ["KMP_BLOCKTIME"] = "30"

os.environ["KMP_SETTINGS"] = "1"

os.environ["KMP_AFFINITY"]= "granularity=fine,verbose,compact,1,0"

我已经定义了以下变量:NUM_PARALLEL_EXEC_UNITS为4。

但是它阻塞在25。

我的目标是提高流程速度。预先感谢。

0 个答案:

没有答案