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:

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 )

Google+ photo

You are commenting using your Google+ 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 )

Connecting to %s

This site uses Akismet to reduce spam. Learn how your comment data is processed.