diff --git a/requirements.txt b/requirements.txt
index df7169d..2144070 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 numpy>=1.16
 tqdm>=4.35.0
 sk-video>=1.1.10
-torch>=1.6.0
+torch=1.7.1
 opencv-python>=4.1.2
 moviepy>=1.0.3
 torchvision