mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-10 12:33:42 +00:00
add PyTorch version
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user