books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added Apple Silicon GPU device update (#820)

Browse Source 
* Added Apple Silicon GPU device

* Added Apple Silicon GPU device

* delete: remove unused model.pth file from understanding-buffers

* update

* update

---------

Co-authored-by: missflash <missflash@gmail.com>
This commit is contained in:
Sebastian Raschka
2025-09-13 12:48:06 -05:00
committed by GitHub
parent 8e170312fe
commit fc101b710e
2 changed files with 840 additions and 819 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ch03/02_bonus_efficient-multihead-attention/mha-implementations.ipynb
View File

@@ -58,6 +58,7 @@
"name": "stdout", "name": "stdout",
"output_type": "stream", "output_type": "stream",
"text": [ "text": [
"Using device: cuda\n",
"PyTorch version: 2.8.0\n" "PyTorch version: 2.8.0\n"
] ]
} }
@@ -66,7 +67,14 @@
"import torch\n", "import torch\n",
"\n", "\n",
"torch.manual_seed(123)\n", "torch.manual_seed(123)\n",
"device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n", "if torch.backends.mps.is_available():\n",
" device = torch.device(\"mps\") # Apple Silicon GPU (Metal)\n",
"elif torch.cuda.is_available():\n",
" device = torch.device(\"cuda\") # NVIDIA GPU\n",
"else:\n",
" device = torch.device(\"cpu\") # CPU fallback\n",
"\n",
"print(f\"Using device: {device}\")\n",
"print(f\"PyTorch version: {torch.__version__}\")\n", "print(f\"PyTorch version: {torch.__version__}\")\n",
"\n", "\n",
"batch_size = 8\n", "batch_size = 8\n",

1649
ch03/03_understanding-buffers/understanding-buffers.ipynb
View File

File diff suppressed because it is too large Load Diff