diff --git a/requirements.txt b/requirements.txt
index d01803a..101b86c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 torch
+torchvision
 numpy
 Pillow
 color-matcher