Add comments on how to enable Caffe GPU operations.

google · Aug 12, 2015 · 4732a3d · 4732a3d
1 parent 08ed923
commit 4732a3d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dream.ipynb b/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": {}
   }
  ]
-}
+}