Fix empty device issue (#904)

2026-04-10 12:33:42 +00:00 · 2025-11-05 20:04:44 -06:00
parent bcc73f731d
commit 58f45ae5a7
6 changed files with 13 additions and 0 deletions
--- a/appendix-E/01_main-chapter-code/appendix-E.ipynb
+++ b/appendix-E/01_main-chapter-code/appendix-E.ipynb
@@ -561,6 +561,8 @@
    "    major, minor = map(int, torch.__version__.split(\".\")[:2])\n",
    "    if (major, minor) >= (2, 9):\n",
    "        device = torch.device(\"mps\")\n",
+    "    else:\n",
+    "        device = torch.device(\"cpu\")\n",
    "else:\n",
    "    device = torch.device(\"cpu\")\n",
    "\n",