diff --git a/requirements.txt b/requirements.txt
index c6ee0c2..ae2f616 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,8 +2,8 @@ flask==1.1.2
 numpy==1.19.4
 pillow==8.0.1
 scikit-image==0.17.2
-torch==1.7.0
-torchvision==0.8.1
+torch~=1.7.0
+torchvision~=0.8.1
 waitress==1.4.4
 tqdm==4.51.0
 requests==2.24.0