Add Google Colab disclaimer

Add a disclaimer to enable GPUs and install the necessary dependencies when running the Notebook in Google Colab. Signed-off-by: Dimitris Poulopoulos <[email protected]>
dpoulopoulos · Sep 10, 2024 · 5043037 · 5043037
1 parent d24bcba
commit 5043037
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -19,7 +19,9 @@ to scale and automate the experiment in a Kubeflow cluster.
   <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
 </a>
 
-<br>
+> **Note**: If you're using Google Colab, make sure to enable the GPU runtime
+> before running the notebook and install the required dependencies, as shown
+> in the `requirements.txt` file.
 
 ## Approach