diff --git a/docs/requirements.txt b/docs/requirements.txt
index 99c41798..4cb316c8 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -4,4 +4,5 @@ sphinx_design
 sphinx_copybutton
 sphinx-tabs
 matplotlib
+torchvision
 -e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme