wip

ami-iit · Jun 26, 2024 · 5861322 · 5861322
1 parent e5eafc6
commit 5861322
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/examples/PD_controller.ipynb b/examples/PD_controller.ipynb
@@ -41,7 +41,6 @@
     "    !apt install xvfb\n",
     "    clear_output()\n",
     "\n",
-    "if IS_COLAB or IS_RTD:\n",
     "    # Start Xvfb and set \"DISPLAY\" environment variable\n",
     "    import pyvirtualdisplay\n",
     "\n",
@@ -163,12 +162,12 @@
     "import subprocess\n",
     "\n",
     "\n",
-    "if IS_COLAB:\n",
+    "if IS_COLAB or IS_RTD:\n",
     "    if subprocess.run(\"ffmpeg -version\", shell=True).returncode:\n",
     "        !command -v ffmpeg >/dev/null || (apt update && apt install -y ffmpeg)\n",
     "        clear_output()\n",
     "\n",
-    "    if subprocess.run(\"nvidia-smi\").returncode:\n",
+    "    if IS_COLAB and subprocess.run(\"nvidia-smi\").returncode:\n",
     "        raise RuntimeError(\n",
     "            \"Cannot communicate with GPU. \"\n",
     "            \"Make sure you are using a GPU Colab runtime. \"\n",