diff --git a/tensorflow/machine_translation.py b/tensorflow/machine_translation.py index 6753a47..4b55389 100644 --- a/tensorflow/machine_translation.py +++ b/tensorflow/machine_translation.py @@ -519,7 +519,11 @@ def do_validataion(): return total_loss / count config = tf.ConfigProto( - intra_op_parallelism_threads=1, inter_op_parallelism_threads=1) + intra_op_parallelism_threads=1, + inter_op_parallelism_threads=1, + log_device_placement=True) + config.gpu_options.allow_growth = True + with tf.Session(config=config) as sess: init_g = tf.global_variables_initializer() init_l = tf.local_variables_initializer() diff --git a/tensorflow/mnist.py b/tensorflow/mnist.py index eda5885..9bf4d08 100644 --- a/tensorflow/mnist.py +++ b/tensorflow/mnist.py @@ -116,7 +116,11 @@ def eval_test(): return g_acc[1] config = tf.ConfigProto( - intra_op_parallelism_threads=1, inter_op_parallelism_threads=1) + intra_op_parallelism_threads=1, + inter_op_parallelism_threads=1, + log_device_placement=True) + config.gpu_options.allow_growth = True + with tf.Session(config=config) as sess: init_g = tf.global_variables_initializer() init_l = tf.local_variables_initializer() diff --git a/tensorflow/resnet50.py b/tensorflow/resnet50.py index 7f4263b..3b643a5 100644 --- a/tensorflow/resnet50.py +++ b/tensorflow/resnet50.py @@ -406,7 +406,13 @@ def test(): print("Pass = %d, Train performance = %f imgs/s, Test accuracy = %f\n" % (pass_id, num_samples / train_elapsed, np.mean(test_accs))) - with tf.Session() as sess: + config = tf.ConfigProto( + intra_op_parallelism_threads=1, + inter_op_parallelism_threads=1, + log_device_placement=True) + config.gpu_options.allow_growth = True + + with tf.Session(config=config) as sess: init_g = tf.global_variables_initializer() init_l = tf.local_variables_initializer() sess.run(init_g) diff --git a/tensorflow/vgg16.py b/tensorflow/vgg16.py index 37197c2..c545ba6 100644 --- a/tensorflow/vgg16.py +++ b/tensorflow/vgg16.py @@ -250,7 +250,13 @@ def test(): })) return np.mean(test_accs) - with tf.Session() as sess: + config = tf.ConfigProto( + intra_op_parallelism_threads=1, + inter_op_parallelism_threads=1, + log_device_placement=True) + config.gpu_options.allow_growth = True + + with tf.Session(config=config) as sess: init_g = tf.global_variables_initializer() init_l = tf.local_variables_initializer() sess.run(init_g)