Updated Colab Script to install CUDA 12.6 and CUDNN

roop-unleashed.ipynb

@@ -16,8 +16,7 @@
         "id": "CanIXgLJgaOj"
       },
       "source": [
-        "Install CUDA 12.6 on Google Cloud Compute\n",
-        "(currently unnecessary because the latest 12.x should be already installed)"
+        "Install CUDA 12.6 & CUDNN on Google Cloud Compute"
       ]
     },
     {
@@ -28,12 +27,15 @@
       },
       "outputs": [],
       "source": [
-        "# don't run this cell if you know that there is at least Cuda 12.4 installed\n",
         "!apt-get -y update\n",
         "!apt-get -y install cuda-toolkit-12-6\n",
+        "!apt-get -y install cudnn9-cuda-12\n",
+        "\n",
         "import os\n",
         "os.environ[\"LD_LIBRARY_PATH\"] += \":\" + \"/usr/local/cuda-12/lib64\"\n",
-        "os.environ[\"LD_LIBRARY_PATH\"] += \":\" + \"/usr/local/cuda-12.6/lib64\""
+        "os.environ[\"LD_LIBRARY_PATH\"] += \":\" + \"/usr/local/cuda-12.6/lib64\"\n",
+        "\n",
+        "!nvcc --version"
       ]
     },
     {
@@ -76,6 +78,10 @@
       },
       "outputs": [],
       "source": [
+        "import torch\n",
+        "print(f\"PyTorch version: {torch.__version__}\")\n",
+        "print(f\"CUDA device is available: {torch.cuda.is_available()}\")\n",
+        "\n",
         "!python run.py"
       ]
     },