tf memory usage 2

imcomking · imcomking · commit e2ca2cfd4a4a · 2016-05-19T17:35:51.000+09:00
diff --git a/Deep_RL_tensorflow/TensorFlow_Tutorial.ipynb b/Deep_RL_tensorflow/TensorFlow_Tutorial.ipynb
@@ -1552,7 +1552,7 @@
    ],
    "source": [
     "# Run! \n",
-    "sess = tf.Session()\n",
+    "sess = tf.Session(config=tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth =True)))\n",
     "# Initialize\n",
     "sess.run(init)    \n",
     "for epoch in range(training_epochs):\n",
