From 6ba97adaee53061d5ee6ccd651a958757cee214b Mon Sep 17 00:00:00 2001
From: rasbt <mail@sebastianraschka.com>
Date: Sat, 9 Mar 2024 17:35:44 -0600
Subject: [PATCH] add PyTorch version

---
 .../mha-implementations.ipynb                                 | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ch03/02_bonus_efficient-multihead-attention/mha-implementations.ipynb b/ch03/02_bonus_efficient-multihead-attention/mha-implementations.ipynb
index 3c6d3bf..85a250e 100644
--- a/ch03/02_bonus_efficient-multihead-attention/mha-implementations.ipynb
+++ b/ch03/02_bonus_efficient-multihead-attention/mha-implementations.ipynb
@@ -26,6 +26,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "PyTorch version: 2.2.1\n",
       "Running on cpu\n"
      ]
     }
@@ -35,6 +36,7 @@
     "\n",
     "torch.manual_seed(123)\n",
     "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
+    "print(f\"PyTorch version: {torch.__version__}\")\n",
     "print(f\"Running on {device}\")\n",
     "\n",
     "batch_size = 8\n",
@@ -855,7 +857,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.6"
+   "version": "3.11.4"
   }
  },
  "nbformat": 4,