update GPU usage section

mrdbourke · Jan 11, 2024 · 0fa794b · 0fa794b
1 parent 1f43ea8
commit 0fa794b
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/00_pytorch_fundamentals.ipynb b/00_pytorch_fundamentals.ipynb
@@ -3505,15 +3505,13 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
+    "### 2.1 Getting PyTorch to run on Apple Silicon\n",
     "\n",
+    "In order to run PyTorch on Apple's M1/M2/M3 GPUs you can use the [`torch.backends.mps`](https://pytorch.org/docs/stable/notes/mps.html) module.\n",
     "\n",
-    "### 2.1 Getting PyTorch to run on the ARM GPUs\n",
+    "Be sure that the versions of the macOS and Pytorch are updated.\n",
     "\n",
-    "In order to run PyTorch on the Apple's M1/M2 GPUs you can use the [`torch.backends.mps`](https://pytorch.org/docs/stable/notes/mps.html) package.\n",
-    "\n",
-    "Be sure that the versions of the MacOS and Pytorch are updated\n",
-    "\n",
-    "You can test if PyTorch has access to a GPU using `torch.backends.mps.is_available()`\n"
+    "You can test if PyTorch has access to a GPU using `torch.backends.mps.is_available()`."
    ]
   },
   {
@@ -3533,9 +3531,9 @@
     }
    ],
    "source": [
-    "# Check for ARM GPU\n",
+    "# Check for Apple Silicon GPU\n",
     "import torch\n",
-    "torch.backends.mps.is_available()"
+    "torch.backends.mps.is_available() # Note this will print false if you're not running on a Mac"
    ]
   },
   {
@@ -3564,7 +3562,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "As before, if the above output `\"mps\"` it means we can set all of our PyTorch code to use the available Apple Arm GPU"
+    "As before, if the above output `\"mps\"` it means we can set all of our PyTorch code to use the available Apple Silicon GPU."
    ]
   },
   {
@@ -3574,11 +3572,11 @@
    "outputs": [],
    "source": [
     "if torch.cuda.is_available():\n",
-    "    device = 'cuda'\n",
+    "    device = \"cuda\" # Use NVIDIA GPU (if available)\n",
     "elif torch.backends.mps.is_available():\n",
-    "    device = 'mps'\n",
+    "    device = \"mps\" # Use Apple Silicon GPU (if available)\n",
     "else:\n",
-    "    device = 'cpu'"
+    "    device = \"cpu\" # Default to CPU if no GPU is available"
    ]
   },
   {