Crash without a GPU

2023-11-05 01:42:13 +01:00 · 2023-11-05 01:42:13 +01:00 · fd24d95981
parent 61a0042c66
commit fd24d95981
1 changed files with 4 additions and 0 deletions
--- a/colab/GPU.ipynb
+++ b/colab/GPU.ipynb
@ -70,6 +70,10 @@
        "Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n",
        "use_google_drive = True #@param {type:\"boolean\"}\n",
        "\n",
+        "import os\n",
+        "if not os.path.isfile(\"/opt/bin/nvidia-smi\"):\n",
+        "  raise RuntimeError(\"⚠️Colab did not give you a GPU due to usage limits, this can take a few hours before they let you back in. Check out https://lite.koboldai.net for a free alternative (that does not provide an API link but can load KoboldAI saves and chat cards) or subscribe to Colab Pro for immediate access.⚠️\")\n",
+        "\n",
        "!nvidia-smi\n",
        "from google.colab import drive\n",
        "if use_google_drive:\n",