fix: Colab Cuda version

SociallyIneptWeeb · Dec 16, 2023 · c90f5ee · c90f5ee
1 parent 536e2c6
commit c90f5ee
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/AICoverGen_colab.ipynb b/AICoverGen_colab.ipynb
@@ -48,6 +48,9 @@
         "timer = 0\n",
         "threading.Thread(target=update_timer_and_print, daemon=True).start()\n",
         "\n",
+        "# Colab Cuda Fix Version",
+        "!sudo apt update\n",
+        "!yes | sudo DEBIAN_FRONTEND=noninteractive apt-get -yq install cuda-11-8\n",
         "!rm -rf sample_data\n",
         "%cd HRVC\n",
         "clear_output()\n",