diff --git a/mnist_keras.py b/mnist_keras.py
index 9fbbb70..6a33f00 100644
--- a/mnist_keras.py
+++ b/mnist_keras.py
@@ -18,8 +18,8 @@
 
 # Horovod: pin GPU to be used to process local rank (one GPU per process)
 config = tf.ConfigProto()
-config.gpu_options.allow_growth = True
-config.gpu_options.visible_device_list = str(hvd.local_rank())
+# config.gpu_options.allow_growth = True
+# config.gpu_options.visible_device_list = str(hvd.local_rank())
 K.set_session(tf.Session(config=config))
 
 batch_size = 128