[docs] Fix tutorial notebook run in Colab with GPU

BioMedIA · Jul 7, 2023 · c47ac8c · c47ac8c
1 parent d51f586
commit c47ac8c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docs/tutorials/pairwise-registration-intro.ipynb b/docs/tutorials/pairwise-registration-intro.ipynb
@@ -47,7 +47,8 @@
     "except ImportError:\n",
     "    if not os.getenv(\"COLAB_RELEASE_TAG\"):\n",
     "        raise\n",
-    "    !git clone --depth=1 https://github.com/BioMedIA/deepali.git && pip install ./deepali"
+    "    !git clone https://github.com/BioMedIA/deepali.git && pip install ./deepali\n",
+    "    from deepali.utils.cli import cuda_visible_devices"
    ]
   },
   {