diff --git a/requirements.txt b/requirements.txt
index 2cf2897..db73297 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,5 +7,5 @@ numpy
 opencv-python
 Pillow
 torch>=1.7
-torchvision
+torchvision==0.16.2
 tqdm