Encouraging TensorFlow to use more cores

I have been trying out some settings to see if I could speed up the training rate in the DQN-tensorflow Breakout demo. Configuring the session with the following code seemed to double the performance on my 5820K/GTX 1070 system (this is an extract from main.py in DQN-tensorflow):

 with tf.Session(config=tf.ConfigProto(
   device_count={"CPU":12},
   inter_op_parallelism_threads=1,
   intra_op_parallelism_threads=1,
   gpu_options=gpu_options,
 )) as sess:
Advertisements

One thought on “Encouraging TensorFlow to use more cores

  1. Pingback: TensorFlow如何充分使用所有CPU核数,提高TensorFlow的CPU使用率,以及Intel的MKL加速 | David 9的博客 --- 不怕"过拟合"

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s