diff --git a/setup.cfg b/setup.cfg
index 1b17fd2..9a2b7bd 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -22,3 +22,4 @@ python_requires = >=3.7
 install_requires =
     numpy
     torch
+    torchvision