enh(demo): Add print for GPU available

daavoo · daavoo · commit b4e50f94c06e · 2025-01-15T16:00:54.000+01:00
diff --git a/demo/notebook.ipynb b/demo/notebook.ipynb
@@ -61,7 +61,9 @@
     "import torch\n",
     "\n",
     "if not torch.cuda.is_available():\n",
-    "    raise RuntimeError(\"GPU not available\")"
+    "    raise RuntimeError(\"GPU not available\")\n",
+    "else:\n",
+    "    print(\"GPU is available!\")"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,9 @@`
`61`	`61`	`"import torch\n",`
`62`	`62`	`"\n",`
`63`	`63`	`"if not torch.cuda.is_available():\n",`
`64`		`- " raise RuntimeError(\"GPU not available\")"`
	`64`	`+ " raise RuntimeError(\"GPU not available\")\n",`
	`65`	`+ "else:\n",`
	`66`	`+ " print(\"GPU is available!\")"`
`65`	`67`	`]`
`66`	`68`	`},`
`67`	`69`	`{`