add PyTorch version

This commit is contained in:
rasbt
2024-03-09 17:35:44 -06:00
parent 1d819c3d9c
commit 6ba97adaee

View File

@@ -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,