From 31c88e8574a5f4ade7020e195933a6c2300d24da Mon Sep 17 00:00:00 2001
From: Matteo Bettini <mb2389@cl.cam.ac.uk>
Date: Thu, 16 Nov 2023 15:18:16 +0100
Subject: [PATCH] [Docs] Fix colab notebook

---
 notebooks/run.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/notebooks/run.ipynb b/notebooks/run.ipynb
index 38ba9d78..24672366 100644
--- a/notebooks/run.ipynb
+++ b/notebooks/run.ipynb
@@ -46,7 +46,7 @@
    "outputs": [],
    "source": [
     "#@title\n",
-    "!pip install -U torch\n",
+    "!pip install -U torch torchvision\n",
     "!pip install -e ."
    ]
   },