Check for GPU for notebook run

GAMSPy_integration_example/trnsport_cuopt.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 32,
+   "execution_count": null,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-07-17T10:21:15.883029Z",
@@ -42,14 +42,66 @@
     }
    ],
    "source": [
-    "# remove -q to debug issues with pip installs\n",
-    "!pip install --upgrade -q --extra-index-url=https://pypi.nvidia.com cuopt-cu12 nvidia-cuda-runtime-cu12==12.8.* nvidia-nvjitlink-cu12\n",
-    "!pip install -q gamspy\n",
+    "# Enable this in case you are running this in google colab or such places where cuOpt is not yet installed\n",
+    "#!pip uninstall -y cuda-python cuda-bindings cuda-core\n",
+    "#!pip install --upgrade --extra-index-url=https://pypi.nvidia.com cuopt-cu12 nvidia-nvjitlink-cu12 rapids-logger==0.1.19\n",
+    "#!pip install --upgrade --extra-index-url=https://pypi.nvidia.com cuopt-cu13 nvidia-nvjitlink-cu13 rapids-logger==0.1.19"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import subprocess\n",
-    "import sys\n",
-    "!wget -nc -nv -O cuopt-link-release.zip \"https://github.com/GAMS-dev/cuoptlink-builder/releases/download/v0.0.1/cuopt-link-release.zip\"\n",
-    "gams_base_path = subprocess.check_output([sys.executable, '-m', 'gamspy', 'show', 'base']).decode('utf-8').strip()\n",
-    "subprocess.run(f\"unzip -o cuopt-link-release.zip -d {gams_base_path}\", shell=True, check=True)"
+    "import html\n",
+    "from IPython.display import display, HTML\n",
+    "\n",
+    "def check_gpu():\n",
+    "    try:\n",
+    "        result = subprocess.run([\"nvidia-smi\"], capture_output=True, text=True, timeout=5)\n",
+    "        result.check_returncode()\n",
+    "        lines = result.stdout.splitlines()\n",
+    "        gpu_info = lines[2] if len(lines) > 2 else \"GPU detected\"\n",
+    "        gpu_info_escaped = html.escape(gpu_info)\n",
+    "        display(HTML(f\"\"\"\n",
+    "        <div style=\"border:2px solid #4CAF50;padding:10px;border-radius:10px;background:#e8f5e9;\">\n",
+    "            <h3>✅ GPU is enabled</h3>\n",
+    "            <pre>{gpu_info_escaped}</pre>\n",
+    "        </div>\n",
+    "        \"\"\"))\n",
+    "        return True\n",
+    "    except (subprocess.CalledProcessError, subprocess.TimeoutExpired, FileNotFoundError, IndexError) as e:\n",
+    "        display(HTML(\"\"\"\n",
+    "        <div style=\"border:2px solid red;padding:15px;border-radius:10px;background:#ffeeee;\">\n",
+    "            <h3>⚠️ GPU not detected!</h3>\n",
+    "            <p>This notebook requires a <b>GPU runtime</b>.</p>\n",
+    "            \n",
+    "            <h4>If running in Google Colab:</h4>\n",
+    "            <ol>\n",
+    "              <li>Click on <b>Runtime → Change runtime type</b></li>\n",
+    "              <li>Set <b>Hardware accelerator</b> to <b>GPU</b></li>\n",
+    "              <li>Then click <b>Save</b> and <b>Runtime → Restart runtime</b>.</li>\n",
+    "            </ol>\n",
+    "            \n",
+    "            <h4>If running in Docker:</h4>\n",
+    "            <ol>\n",
+    "              <li>Ensure you have <b>NVIDIA Docker runtime</b> installed (<code>nvidia-docker2</code>)</li>\n",
+    "              <li>Run container with GPU support: <code>docker run --gpus all ...</code></li>\n",
+    "              <li>Or use: <code>docker run --runtime=nvidia ...</code> for older Docker versions</li>\n",
+    "              <li>Verify GPU access: <code>docker run --gpus all nvidia/cuda:12.0.0-base-ubuntu22.04 nvidia-smi</code></li>\n",
+    "            </ol>\n",
+    "            \n",
+    "            <p><b>Additional resources:</b></p>\n",
+    "            <ul>\n",
+    "              <li><a href=\"https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html\" target=\"_blank\">NVIDIA Container Toolkit Installation Guide</a></li>\n",
+    "            </ul>\n",
+    "        </div>\n",
+    "        \"\"\"))\n",
+    "        return False\n",
+    "\n",
+    "check_gpu()"
    ]
   },
   {