.

2024-06-01 23:58:51 -04:00
parent 524c12e7aa
commit d44fc8e177
25 changed files with 2951 additions and 39 deletions
@@ -10,8 +10,7 @@
     "name": "stdout",
     "output_type": "stream",
     "text": [
-      "CUDA is available. GPU is ready for use.\n",
-      "Number of GPUs: 1\n",
+      "CUDA is available. Number of GPUs: 1\n",
      "GPU Name: NVIDIA GeForce RTX 3060\n"
     ]
    }
@@ -21,8 +20,7 @@
    "\n",
    "# Check if CUDA is available\n",
    "if torch.cuda.is_available():\n",
-    "    print(\"CUDA is available. GPU is ready for use.\")\n",
-    "    print(f\"Number of GPUs: {torch.cuda.device_count()}\")\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",
    "else:\n",
    "    print(\"CUDA is not available. No GPU detected.\")\n"
@@ -41,7 +39,7 @@
  {
   "cell_type": "code",
   "execution_count": null,
-   "id": "53cd2a47-ef77-401a-8497-8fc920919f22",
+   "id": "7313a620-a0eb-4207-a12a-90aeee3cd980",
   "metadata": {},
   "outputs": [],
   "source": []