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 ."
    ]
   },