From c47ac8cf81ce0978b3095d79592fcd1f017e23c6 Mon Sep 17 00:00:00 2001 From: Andreas Schuh Date: Fri, 7 Jul 2023 01:52:43 +0000 Subject: [PATCH] [docs] Fix tutorial notebook run in Colab with GPU --- docs/tutorials/pairwise-registration-intro.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/tutorials/pairwise-registration-intro.ipynb b/docs/tutorials/pairwise-registration-intro.ipynb index 9b8495f..5a53861 100644 --- 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" ] }, {