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

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