Disable CUDA

cemac · Sep 25, 2024 · 1a9a05d · 1a9a05d
1 parent d897bb2
commit 1a9a05d
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/Image_Segmentation.ipynb b/Image_Segmentation.ipynb
@@ -180,7 +180,22 @@
     "import xarray as xr\n",
     "import matplotlib.pyplot as plt\n",
     "import os\n",
-    "import pickle"
+    "import pickle\n",
+    "import fastai"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c20c5fd0",
+   "metadata": {
+    "vscode": {
+     "languageId": "julia"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "fastai.torch_core.default_device(use_cuda=False)"
    ]
   },
   {