progress

pytorch/notebooks/Testing GPU.ipynb

@@ -10,7 +10,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 2,
    "id": "6b269e64-be58-43b5-ad60-0fbd1d37861a",
    "metadata": {},
    "outputs": [
@@ -19,7 +19,8 @@
      "output_type": "stream",
      "text": [
       "CUDA is available. Number of GPUs: 1\n",
-      "GPU Name: NVIDIA GeForce RTX 3060\n"
+      "GPU Name: NVIDIA GeForce RTX 3060\n",
+      "GPU Capability: 8\n"
      ]
     }
    ],
@@ -30,6 +31,7 @@
     "if torch.cuda.is_available():\n",
     "    print(f\"CUDA is available. Number of GPUs: {torch.cuda.device_count()}\")\n",
     "    print(f\"GPU Name: {torch.cuda.get_device_name(0)}\")\n",
+    "    print(f\"GPU Capability: {torch.cuda.get_device_capability()[0]}\")\n",
     "else:\n",
     "    print(\"CUDA is not available. No GPU detected.\")\n"
    ]