dream.ipynb

@@ -44,6 +44,11 @@
       "\n",
       "import caffe\n",
       "\n",
+      "# If your GPU supports CUDA and Caffe was built with CUDA support,\n",
+      "# uncomment the following to run Caffe operations on the GPU.\n",
+      "# caffe.set_mode_gpu()\n",
+      "# caffe.set_device(0) # select GPU device if multiple devices exist\n",
+      "\n",
       "def showarray(a, fmt='jpeg'):\n",
       "    a = np.uint8(np.clip(a, 0, 255))\n",
       "    f = StringIO()\n",
@@ -570,4 +575,4 @@
    "metadata": {}
   }
  ]
-}
+}