added pytorch installation check

BiAPoL · Aug 19, 2024 · b1fc29e · b1fc29e
1 parent d2b51dd
commit b1fc29e
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/docs/johannes_mueller/yolo_from_omero/train_yolo.ipynb b/docs/johannes_mueller/yolo_from_omero/train_yolo.ipynb
@@ -99,6 +99,23 @@
     "from sklearn.model_selection import train_test_split"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "We check whether pytorch has been installed correctly:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import torch\n",
+    "torch.cuda.is_available()"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 54,